2022&2023华为OD机试 - 竖直四子棋（Python）

Posted 2023-03-04 坚果的博客

竖直四子棋

题目

竖直四子棋的棋盘是竖立起来的，双方轮流选择棋盘的一列下子，
棋子因重力落到棋盘底部或者其他棋子之上，当一列的棋子放满时，无法再在这列上下子。
一方的4个棋子横、竖或者斜方向连成一线时获胜。
现给定一个棋盘和红蓝对弈双方的下子步骤，判断红方或蓝方是否在某一步获胜。
下面以一个6x5的棋盘图示说明落子过程：
下面给出横、竖和斜方向四子连线的图示：

输入

输入为 2 行，第一行指定棋盘的宽和高，为空格分隔的两个数字；
第二行依次间隔指定红蓝双方的落子步骤，第 1 步为红方的落子，第 2 步为蓝方的落子，第 3 步为红方的落子，以此类推。
步骤由空格分隔的一组数字表示，每个数字为落子的列的编号（最左边的列编号为 1，往右递增）。用例保证数字均为 32 位有符号数。

输出

如果落子过程中红方获胜，输出 N,red ；
如果落子过程中蓝方获胜，输出 N,blue ；
如果出现非法的落子步骤，输出 N,error。
N为落子步骤的序号，从1开始。如果双方都没有获胜，输出 0,draw 。
非法落子步骤有两种，一是列的编号超过棋盘范围