Sudoku.py

#!/usr/bin/env python
def initiate():
    box.append([0, 1, 2, 9, 10, 11, 18, 19, 20])
    box.append([3, 4, 5, 12, 13, 14, 21, 22, 23])
    box.append([6, 7, 8, 15, 16, 17, 24, 25, 26])
    box.append([27, 28, 29, 36, 37, 38, 45, 46, 47])
    box.append([30, 31, 32, 39, 40, 41, 48, 49, 50])
    box.append([33, 34, 35, 42, 43, 44, 51, 52, 53])
    box.append([54, 55, 56, 63, 64, 65, 72, 73, 74])
    box.append([57, 58, 59, 66, 67, 68, 75, 76, 77])
    box.append([60, 61, 62, 69, 70, 71, 78, 79, 80])
    for i in range(0, 81, 9):
        row.append(range(i, i+9))
    for i in range(9):
        column.append(range(i, 80+i, 9))

def valid(n, pos):
    current_row = pos/9
    current_col = pos%9
    current_box = (current_row/3)*3 + (current_col/3)
    for i in row[current_row]:
        if (grid[i] == n):
            return False
    for i in column[current_col]:
        if (grid[i] == n):
            return False
    for i in box[current_box]:
        if (grid[i] == n):
            return False
    return True

def solve():
    i = 0
    proceed = 1
    diff = 0.0
    while(i < 81):
        if given[i]:
            if proceed:
                    i += 1
            else:
                i -= 1
        else:
            n = grid[i]
            prev = grid[i]
            while(n < 9):
              if (n < 9):
                  n += 1
              if valid(n, i):
                  grid[i] = n
                  proceed = 1
                  break
            if (grid[i] == prev):
               grid[i] = 0
               proceed = 0
            if proceed:
               i += 1
            else:
               i -=1
               diff = diff + 1.0
    return diff
def gotonext():
    global now_selected
    now_selected = (now_selected+1) % 81
def checkall():
    for i in xrange(81):
        if not valid(grid[i],i):
            return False
    return True

grid = [0]*81
given = [False]*81
box = []
row = []
column = []
initiate()
diff = 0.0
import pygame
import sys
pygame.init()
size = 600, 732
black = 0, 0, 0
screen = pygame.display.set_mode(size)
pygame.display.set_caption('Sudoku Solver')
template = pygame.image.load("SUDOKU.jpg")
rect = template.get_rect()
font = pygame.font.Font('Font.ttf', 48)
now_selected = 0
running = True
while running:
    screen.blit(template, rect)
    for i in range(81):
        if grid[i]:
        if given[i]:
            color = (0,0,255)
        else:
            color = (0,0,0)
        text = font.render(str(grid[i]), 1, color)
            ro = i/9
            col = i%9
            textpos = text.get_rect(centerx = col*66 + 33, centery = ro*66 + 33)
            screen.blit(text, textpos)
    try:
        screen.blit(font.render('_', 1, (150,150,150)), font.render('O', 1, (100,100,100)).get_rect(centerx = (now_selected%9)*66 + 33, centery = (now_selected/9)*66 + 33))
    except:
        pass
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
        elif event.type == pygame.MOUSEBUTTONDOWN:
        x,y = pygame.mouse.get_pos()
        col_edit = x/66
        ro_edit = y/66
        last_selected = now_selected
        now_selected = ro_edit*9 + col_edit
        if (now_selected > 80):
            gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_1 or event.key == pygame.K_KP1) :
            if valid(1,now_selected): grid[now_selected] = 1;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_2 or event.key == pygame.K_KP2) :
            if valid(2,now_selected): grid[now_selected] = 2;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_3 or event.key == pygame.K_KP3) :
            if valid(3,now_selected): grid[now_selected] = 3;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_4 or event.key == pygame.K_KP4)  :
            if valid(4,now_selected): grid[now_selected] = 4;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_5 or event.key == pygame.K_KP5) :
            if valid(5,now_selected): grid[now_selected] = 5;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_6 or event.key == pygame.K_KP6) :
            if valid(6,now_selected): grid[now_selected] = 6;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_7 or event.key == pygame.K_KP7) :
            if valid(7,now_selected): grid[now_selected] = 7;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_8 or event.key == pygame.K_KP8) :
            if valid(8,now_selected): grid[now_selected] = 8;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_9 or event.key == pygame.K_KP9) :
            if valid(9,now_selected): grid[now_selected] = 9;given[now_selected]=True;gotonext()
        elif event.type == pygame.KEYDOWN and (event.key == pygame.K_0 or event.key == pygame.K_BACKSPACE) :
            grid[now_selected] = 0;given[now_selected]=False;
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_UP:
        now_selected = (now_selected-9) % 81
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_DOWN:
        now_selected = (now_selected+9) % 81
    elif event.type == pygame.KEYDOWN and (event.key == pygame.K_RIGHT or event.key == pygame.K_TAB):
        now_selected = (now_selected+1) % 81
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_LEFT:
        now_selected = (now_selected-1) % 81
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_RETURN:
        for i in xrange(81):
            if not given[i]:
                grid[i]=0
        solve()
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_DELETE:
        grid = [0]*81
        given = [False]*81
    elif event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
        running = False
    try:
        pygame.display.update()
    except:
        pass
pygame.quit()