关注
刚开始也是90%,后来发现有个边界出了问题,改完就ac了 def up(a):
# 向上合并, 1
for i in range(3):
for j in range(4):
if a[i][j] == a[i + 1][j]:
a[i][j] = 2*a[i][j]
a[i + 1][j] = 0
for j in range(4):
idxs = []
for i in range(4):
if a[i][j] == 0:
idxs.append(i)
elif a[i][j] != 0 and len(idxs) >0:
idx = idxs.pop(0)
a[i][j],a[idx][j] = a[idx][j],a[i][j]
idxs.append(i)
return a
def low(a):
# 向下合并, 2
for i in range(3,0,-1):
for j in range(4):
if a[i][j] == a[i - 1][j]:
a[i][j] = 2*a[i][j]
a[i - 1][j] = 0
for j in range(4):
idxs = []
for i in range(3,-1,-1):
if a[i][j] == 0:
idxs.append(i)
elif a[i][j] != 0 and len(idxs) >0:
idx = idxs.pop(0)
a[i][j],a[idx][j] = a[idx][j],a[i][j]
idxs.append(i)
return a
def left(a):
#向左合并, 3
for i in range(4):
for j in range(3):
if a[i][j] == a[i][j+1]:
a[i][j] = 2*a[i][j]
a[i][j+1] =0
# 去除0
for i in range(4):
idxs = []
for j in range(4):
if a[i][j] == 0:
idxs.append(j)
elif a[i][j] != 0 and len(idxs) >0:
idx = idxs.pop(0)
a[i][j],a[i][idx] = a[i][idx],a[i][j]
idxs.append(j)
return a
def right(a):
for i in range(4):
for j in range(3,0,-1):
if a[i][j] == a[i][j-1]:
a[i][j] = 2*a[i][j]
a[i][j-1] =0
#去零
for i in range(4):
idxs = []
for j in range(3,-1,-1):
if a[i][j] == 0:
idxs.append(j)
elif a[i][j] != 0 and len(idxs) >0:
idx = idxs.pop(0)
a[i][j],a[i][idx] = a[i][idx],a[i][j]
idxs.append(j)
return a
dx = int(input())
a = []
for _ in range(4):
a.append(list(map(int, input().split())))
if dx == 1:
res = up(a)
elif dx ==2:
res = low(a)
elif dx ==3:
res = left(a)
elif dx == 4:
res = right(a)
for item in res:
print(' '.join('%s' %i for i in item))
查看原帖
点赞 评论
相关推荐
牛客热帖
更多
正在热议
更多
# 从顶到拉给所有面过的公司评分 #
12179次浏览 114人参与
# 机械人春招想让哪家公司来捞你? #
356768次浏览 3104人参与
# 为了求职,我做过的疯狂伪装 #
10140次浏览 160人参与
# 晒晒你的中秋福利 #
14663次浏览 91人参与
# 职场破冰,你们都聊什么? #
5625次浏览 57人参与
# 工作压力大怎么缓解 #
104672次浏览 1048人参与
# 机械人怎么评价今年的华为 #
208470次浏览 1524人参与
# 广联达求职进展汇总 #
10594次浏览 50人参与
# bilibili求职进展汇总 #
84103次浏览 773人参与
# 大家实习每天都在干啥 #
88586次浏览 517人参与
# 你面试被问到过哪些不会的问题? #
18117次浏览 714人参与
# 聊聊这家公司值得去吗 #
552556次浏览 3676人参与
# 实习要如何选择和准备? #
114393次浏览 1436人参与
# 秋招报数:你投了多少家公司? #
25811次浏览 259人参与
# 上班后和你想的一样吗? #
79069次浏览 630人参与
# 电网笔面经互助 #
46337次浏览 428人参与
# 秋招的嫡长offer #
24720次浏览 236人参与
# 你觉得早上几点上班合适? #
82225次浏览 329人参与
# 上班摸鱼,你都在干些什么? #
5871次浏览 102人参与
# 秋招OC许愿 #
345388次浏览 2521人参与