結果

問題 No.2112 All 2x2 are Equal
ユーザー とりゐ
提出日時 2022-10-28 22:50:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 297 ms / 2,000 ms
コード長 1,547 bytes
コンパイル時間 315 ms
コンパイル使用メモリ 82,416 KB
実行使用メモリ 93,020 KB
最終ジャッジ日時 2024-07-06 02:03:34
合計ジャッジ時間 9,402 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

def calc(h,w):
  # w is odd
  ans=[[0]*w for i in range(h)]
  for i in range(h):
    for j in range(w):
      if i%2==0:
        ans[i][j]=j+1
      else:
        ans[i][j]=w-j
  
  for i in range((h+1)//2):
    if 2*i+1!=h:
      i1=2*i
      i2=i1+1
      for j in range(w):
        if j%2==0:
          ans[i1][j]+=w*i
          ans[i2][j]+=w*(h-1-i)
        else:
          ans[i2][j]+=w*i
          ans[i1][j]+=w*(h-1-i)  
    else:
      for j in range(w):
        ans[-1][j]+=w*(h//2)
  return ans

def calc2(h,w):
  # w and h are even
  ans=[[0]*w for i in range(h)]
  for i in range(h):
    for j in range(w):
      if i%2==0:
        ans[i][j]=j+1
      else:
        ans[i][j]=w-j
  
  for i in range(h//2):
    i1=2*i
    i2=2*i+1
    for j in range(w//2):
      if j%2==0:
        ans[i1][j]+=w*i
        ans[i2][j]+=w*(h-1-i)
      else:
        ans[i2][j]+=w*i
        ans[i1][j]+=w*(h-1-i)
    for j in range(w//2,w):
      if j%2==0:
        ans[h-1-i1][j]+=w*i
        ans[h-1-i2][j]+=w*(h-1-i)
      else:
        ans[h-1-i2][j]+=w*i
        ans[h-1-i1][j]+=w*(h-1-i)
  
  return ans

h,w=map(int,input().split())
if w%2==1:
  ans=calc(h,w)
  
elif h%2==1:
  tmp=calc(w,h)
  ans=[[0]*w for i in range(h)]
  for i in range(h):
    for j in range(w):
      ans[i][j]=tmp[j][i]

else:
  ans=calc2(h,w)

print('Yes')
for i in ans:
  print(*i)

s=set()
for i in range(h-1):
  for j in range(w-1):
    tmp=0
    for di in range(2):
      for dj in range(2):
        tmp+=ans[i+di][j+dj]
    s.add(tmp)

if len(s)!=1:
  raise Exception
0