結果

問題 No.2797 Square Tile
ユーザー hitonanodehitonanode
提出日時 2024-06-29 01:39:16
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 120 ms / 2,000 ms
コード長 947 bytes
コンパイル時間 268 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 21,760 KB
最終ジャッジ日時 2024-06-29 01:39:20
合計ジャッジ時間 3,080 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 27 ms
10,880 KB
testcase_01 AC 26 ms
10,752 KB
testcase_02 AC 25 ms
10,880 KB
testcase_03 AC 25 ms
10,880 KB
testcase_04 AC 120 ms
21,760 KB
testcase_05 AC 114 ms
20,224 KB
testcase_06 AC 69 ms
16,256 KB
testcase_07 AC 28 ms
11,008 KB
testcase_08 AC 41 ms
11,776 KB
testcase_09 AC 68 ms
15,744 KB
testcase_10 AC 92 ms
18,432 KB
testcase_11 AC 60 ms
14,464 KB
testcase_12 AC 66 ms
15,104 KB
testcase_13 AC 34 ms
11,392 KB
testcase_14 AC 71 ms
16,000 KB
testcase_15 AC 51 ms
12,416 KB
testcase_16 AC 42 ms
12,160 KB
testcase_17 AC 36 ms
11,776 KB
testcase_18 AC 33 ms
11,392 KB
testcase_19 AC 67 ms
15,488 KB
testcase_20 AC 51 ms
12,928 KB
testcase_21 AC 97 ms
14,720 KB
testcase_22 AC 36 ms
12,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import math

def solve(A: int, B: int):
    L = A * A + B * B

    retA: list[tuple[int, int]] = []
    retB: list[tuple[int, int]] = []

    for idx in range(L):
        if A > B:
            retA.append((B * idx % L, A * idx % L))
            retB.append((B * idx % L, A * (idx + 1) % L))
        else:
            retA.append((A * idx % L, B * (idx + 1) % L))
            retB.append((A * idx % L, B * idx % L))
    
    return retA, retB

A, B = map(int, input().split())

G = math.gcd(A, B)

tmpA, tmpB = solve(A // G, B // G)

# print(A, B, G, tmpA, tmpB)

retA: list[tuple[int, int]] = []
retB: list[tuple[int, int]] = []

s = (A * A + B * B) // G

for x in range(G):
    for y in range(G):
        for a, b in tmpA:
            retA.append((a * G + x * s, b * G + y * s))
        for a, b in tmpB:
            retB.append((a * G + x * s, b * G + y * s))

for a, b in sorted(retA):
    print(a, b)
for a, b in sorted(retB):
    print(a, b)
0