結果
問題 |
No.2358 xy+yz+zx=N
|
ユーザー |
![]() |
提出日時 | 2025-03-09 21:52:31 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 928 bytes |
コンパイル時間 | 758 ms |
コンパイル使用メモリ | 82,228 KB |
実行使用メモリ | 84,172 KB |
最終ジャッジ日時 | 2025-03-09 21:52:38 |
合計ジャッジ時間 | 6,629 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 1 WA * 4 TLE * 1 -- * 4 |
ソースコード
from itertools import count, permutations def distinct_permutations(a: list): # -> Generator[list, None, None]: n = len(a) xs = a.copy() while 1: yield xs p = -1 for i in range(n-1): if xs[i] < xs[i+1]: p = i if p == -1: return q = -1 for i in range(p+1, n): if xs[p] < xs[i]: q = i xs[p], xs[q] = xs[q], xs[p] xs[p+1:] = reversed(xs[p+1:]) N = int(input()) s = set() for x in count(): if 3*x*x > N: break for y in count(x): if x*x + 2*y*y > N: break for z in count(y): if x == 0 and y == 0: break t = x*y + y*z + z*x if t > N: break if t == N: s.add((x, y, z)) ans = [] for e in s: for xs in distinct_permutations(list(e)): ans.append(xs[:]) print(len(ans)) for a in ans: print(*a)