結果
| 問題 | No.3567 Modulo Grid |
| コンテスト | |
| ユーザー |
titia
|
| 提出日時 | 2026-06-14 04:04:00 |
| 言語 | PyPy3 (7.3.17) |
| 結果 |
AC
|
| 実行時間 | 493 ms / 2,000 ms |
| コード長 | 5,415 bytes |
| 記録 | |
| コンパイル時間 | 351 ms |
| コンパイル使用メモリ | 85,120 KB |
| 実行使用メモリ | 301,808 KB |
| 最終ジャッジ日時 | 2026-06-14 04:04:06 |
| 合計ジャッジ時間 | 6,208 ms |
|
ジャッジサーバーID (参考情報) |
judge3_0 / judge2_1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 25 |
ソースコード
import sys
input = sys.stdin.readline
# エラトステネスの篩を用いた素因数分解・約数列挙
MAX=3*10**5+10 # 使いたい最大値を指定
# Sieve[i]で、iの最も小さい1でない約数を返す。
Sieve=[i for i in range(MAX)]
for i in range(2,MAX):
if Sieve[i]!=i:
continue
for j in range(i,MAX,i):
if Sieve[j]==j:
Sieve[j]=i
# 素因数分解
def fact(x):
D=dict()
while x!=1:
k=Sieve[x]
if k in D:
D[k]+=1
else:
D[k]=1
x//=k
return D
# 約数列挙
def faclist(x):
LIST=[1]
while x!=1:
k=Sieve[x]
count=0
while x%k==0:
count+=1
x//=k
LIST2=[]
for l in LIST:
for i in range(1,count+1):
LIST2.append(l*k**i)
LIST+=LIST2
return LIST
H,W=list(map(int,input().split()))
flag=0
if H<W:
flag=1
H,W=W,H
F=fact(H*W)
FLIST=faclist(H*W)
FLIST.sort()
A=list(range(1,H*W+1))
LIST=[[] for i in range(len(FLIST)+1)]
for i in range(len(FLIST)-1,-1,-1):
x=FLIST[i]
LIST1=[]
B=[]
for a in A:
if a%x==0:
LIST1.append(a)
else:
B.append(a)
LIST[i]=(LIST1)
A=B
need=0
for i in range(1,len(LIST)):
need+=(len(LIST[i])+W-1)//W
if H>=need+len(LIST)-1:
F=fact(H*W)
NOW=[1]
for f in F:
k=F[f]
X=[1]
for i in range(k):
X.append(X[-1]*f)
X.reverse()
NOW2=[]
for x in X:
L=[]
for a in NOW:
L.append(a*x)
L.reverse()
NOW2.append(L)
NOW2.reverse()
NOW=[]
for x in NOW2:
NOW+=x
#print(NOW)
NOW.reverse()
D={FLIST[i]:i for i in range(len(FLIST))}
ANS=[[0]*W for i in range(H)]
height=0
for c in NOW:
w=0
L=D[c]
if c==1:
height+=1
continue
#print("!",LIST[L])
for x in LIST[L]:
if ANS[height][w]!=0:
height+=1
ANS[height][w]=x
w+=1
if w==W:
w=0
height+=2
ind=0
for i in range(H):
for j in range(W):
if ANS[i][j]==0:
ANS[i][j]=LIST[0][ind]
ind+=1
if flag:
ANS2=[[0]*H for i in range(W)]
for i in range(W):
for j in range(H):
ANS2[i][j]=ANS[j][i]
ANS=ANS2
for ans in ANS:
print(*ans)
elif H>=need:
F=fact(H*W)
NOW=[1]
for f in F:
k=F[f]
X=[1]
for i in range(k):
X.append(X[-1]*f)
X.reverse()
NOW2=[]
for x in X:
L=[]
for a in NOW:
L.append(a*x)
L.reverse()
NOW2.append(L)
NOW2.reverse()
NOW=[]
for x in NOW2:
NOW+=x
#print(NOW)
NOW.reverse()
D={FLIST[i]:i for i in range(len(FLIST))}
ANS=[[0]*W for i in range(H)]
height=0
for c in NOW:
w=0
L=D[c]
if c==1:
height+=1
continue
#print("!",LIST[L])
for x in LIST[L]:
if ANS[height][w]!=0:
height+=1
ANS[height][w]=x
w+=1
if w==W:
w=0
ind=0
for i in range(H):
for j in range(W):
if ANS[i][j]==0:
ANS[i][j]=LIST[0][ind]
ind+=1
if flag:
ANS2=[[0]*H for i in range(W)]
for i in range(W):
for j in range(H):
ANS2[i][j]=ANS[j][i]
ANS=ANS2
for ans in ANS:
print(*ans)
else:
a=H*W
P=[]
for i in range(2,H*W+1):
if a%i==0:
P.append(i)
while a%i==0:
a//=i
P.sort(reverse=True)
A=list(range(1,H*W+1))
LIST=[]
for i in range(len(P)):
if i+1<len(P):
x=P[i]
y=P[i+1]
LIST1=[]
LIST2=[]
B=[]
for a in A:
if a%(x*y)==0:
LIST2.append(a)
elif a%x==0:
LIST1.append(a)
else:
B.append(a)
A=B
LIST.append(LIST1)
LIST.append(LIST2)
else:
x=P[i]
LIST1=[]
B=[]
for a in A:
if a%x==0:
LIST1.append(a)
else:
B.append(a)
LIST.append(LIST1)
A=B
ANS=[[0]*W for i in range(H)]
height=0
for L in LIST:
w=0
for x in L:
if ANS[height][w]!=0:
height+=1
ANS[height][w]=x
w+=1
if w==W:
w=0
ind=0
for i in range(H):
for j in range(W):
if ANS[i][j]==0:
ANS[i][j]=A[ind]
ind+=1
if flag:
ANS2=[[0]*H for i in range(W)]
for i in range(W):
for j in range(H):
ANS2[i][j]=ANS[j][i]
ANS=ANS2
for ans in ANS:
print(*ans)
titia