ADD file via upload

2 years ago · 3bb0b8bc75
parent 24fa4f595b
commit 3bb0b8bc75
1 changed files with 77 additions and 0 deletions
--- a/五子棋.py
+++ b/五子棋.py
@ -0,0 +1,77 @@
+import pygame
+pygame.init()
+screen = pygame.display.set_mode((600, 600)) 
+pygame.display.set_caption('五子棋') 
+space = 20  
+cell_size = 40 
+cell_num = 15 
+chess_arr = [] 
+def get_one_dire_num(lx, ly, dx, dy, m):
+    tx = lx
+    ty = ly
+    s = 0
+    while True:
+        tx += dx
+        ty += dy
+        if tx < 0 or tx >= cell_num or ty < 0 or ty >= cell_num or m[ty][tx] == 0: return s
+        s+=1
+def check_win(chess_arr, flag):
+    m = [[0] * cell_num for i in range(cell_num)]  
+    for x, y, c in chess_arr:
+        if c == flag:
+            m[y][x] = 1  
+    lx = chess_arr[-1][0] 
+    ly = chess_arr[-1][1]  
+    dire_arr = [[(-1, 0), (1, 0)], [(0, -1), (0, 1)], [(-1, -1), (1, 1)],
+                [(-1, 1), (1, -1)]]
+    for dire1, dire2 in dire_arr:
+        dx, dy = dire1
+        num1 = get_one_dire_num(lx, ly, dx, dy, m)
+        dx, dy = dire2
+        num2 = get_one_dire_num(lx, ly, dx, dy, m)
+        if num1 + num2 + 1 >= 5: return True
+    return False
+flag = 1 
+game_state = 1 
+while True: 
+    for event in pygame.event.get(): 
+        if event.type == pygame.QUIT: 
+            pygame.quit() 
+            exit() 
+        if game_state == 1 and event.type == pygame.MOUSEBUTTONUP:
+            x, y = pygame.mouse.get_pos()
+            xi = int(round((x - space) * 1.0 / cell_size)) 
+            yi = int(round((y - space) * 1.0 / cell_size))  
+            if xi >= 0 and xi < cell_num and yi >= 0 and yi < cell_num and (xi, yi,1) not in chess_arr and (xi, yi,2) not in chess_arr:
+                chess_arr.append((xi, yi,flag))
+                if check_win(chess_arr, flag):
+                    game_state = 2 if flag == 1 else 3
+                else:
+                    flag = 2 if flag == 1 else 1
+    screen.fill((204, 153, 102))
+    for x in range(0, cell_size * cell_num, cell_size):
+        pygame.draw.line(screen, (200, 200, 200), (x + space, 0 + space),
+                         (x + space, cell_size * (cell_num - 1) + space), 1)
+    for y in range(0, cell_size * cell_num, cell_size):
+        pygame.draw.line(screen, (200, 200, 200), (0 + space, y + space),
+                         (cell_size * (cell_num - 1) + space, y + space), 1)
+    for x, y,c in chess_arr:
+        chess_color = (30, 30, 30) if c == 1 else (225, 225, 225)
+        pygame.draw.circle(screen, chess_color, [x * cell_size + space, y * cell_size + space], 16, 16)
+    if (4, 4,1) not in chess_arr and (4, 4,2) not in chess_arr:
+        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 4 + space, cell_size * 4 + space), 2, 1)
+    if (10, 10, 1) not in chess_arr and (10, 10, 2) not in chess_arr:
+        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 10 + space, cell_size * 10 + space), 2, 1)
+    if (10, 4, 1) not in chess_arr and (10, 4, 2) not in chess_arr:
+        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 10 + space, cell_size * 4 + space), 2, 1)
+    if (4, 10, 1) not in chess_arr and (4 ,10, 2) not in chess_arr:
+        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 4 + space, cell_size * 10 + space), 2, 1)
+    if (7, 7, 1) not in chess_arr and (7, 7, 2) not in chess_arr:
+        pygame.draw.circle(screen, (0, 0, 0), (cell_size * 7 + space, cell_size * 7 + space), 2, 1)
+    if game_state != 1:
+        myfont = pygame.font.Font(None, 60)
+        white = 210, 210, 0
+        win_text = "%s win" % ("black" if game_state == 2 else "white")
+        textImage = myfont.render(win_text, True, white)
+        screen.blit(textImage, (260, 320))
+    pygame.display.update()