結果

問題 No.3506 All Distance is Square Number
コンテスト
ユーザー Kurao
提出日時 2026-04-18 14:59:55
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 3,522 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 133 ms
コンパイル使用メモリ 85,204 KB
実行使用メモリ 84,080 KB
最終ジャッジ日時 2026-04-18 15:00:00
合計ジャッジ時間 4,376 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#--------------------------------------------------
import sys
import functools
#sys.setrecursionlimit(10**9)
#codon↓--------------------------------------------
_factorial=[1]
def factorial(n):
    while len(_factorial)<=n:
        _factorial.append((_factorial[-1]*len(_factorial))%mod)
    return _factorial[n]

def binom(n,r):
    if r>=mod:
        raise ValueError("r is too big")
    if n<0:
        return 0
    if r>n:
        return 0
    if r<0:
        return 0
    ans=((factorial(n)*pow(factorial(r),mod-2,mod))%mod*pow(factorial(n-r),mod-2,mod))%mod
    return ans

import string
import itertools
alp_low=list(string.ascii_lowercase)
alp_up=list(string.ascii_uppercase)
dij=[[0,1],[1,0],[0,-1],[-1,0]]

mod=998244353
INF=10**18
def dijkstra(edges, num_node,start):
    """

    [node_num,weight>0][
    
    (example)

    Edges = [
        [[1, 4], [2, 3]],
        [[0, 1], [3, 1]],
        [[3, 2]],
        [],
    ]
    """
    import heapq;n=[INF]*num_node;n[start]=0;n_name=[];heapq.heappush(n_name,[0,start])
    while len(n_name):
        _,min_p=heapq.heappop(n_name)
        for f in edges[min_p]:
            g,c=f
            if n[min_p]+c<n[g]:n[g]=n[min_p]+c;heapq.heappush(n_name,[n[min_p]+c,g])
    return n

def nin():
    return list(map(int,input().split()))
def deq(x):
    return [i-1 for i in x]
def pop_cnt(n):
    ans=0
    while n:
        if n%2:
            ans+=1
        n//=2
    return ans

import random
random.seed(0)
def main():
    n,=nin()
    if n==2:
        print(1)
        print(1,2,1)
        print(1,1)
        return
    if n==3:
        print(3)
        print(1,2,1)
        print(1,3,4)
        print(2,3,9)
        print(1,1)
        print(1,2)
        print(1,3)
        return
    if n==4:
        print(5)
        print(1,2,9)
        print(2,3,16)
        print(1,4,1)
        print(2,4,4)
        print(3,4,25)

        print(1,1)
        print(2,1,2)
        print(1,3)
        print(1,2)
        print(1,4)
        print(1,5)
        return
    if 5<=n<=13:
        print("OMG")
        return
    if 14<=n<16:
        size=(n-4)//2
    else:
        size=6
    w2=[1,2,3,5,4,8,6,14,7,23,9,41][:2*size]
    use=list(set(range(1,201))^set(w2))
    use.sort(reverse=True)
    ws=[]
    print(n,n+2*size+1)
    print(1,2,use[-1])
    ws.append(use.pop(-1))
    print(1,2,use[-1])
    ws.append(use.pop(-1))
    base=ws[0]+ws[1]
    for i in range(size):
        print(i%size+2,(i+1)%size+2,w2[i*2])
        print(i%size+2,(i+1)%size+2,w2[i*2+1])
        base+=w2[i*2]
        ws.append(w2[i*2])
        ws.append(w2[i*2+1])
    for i in range(2,n+1):
        print(1,i,use[-1])
        ws.append(use.pop(-1))
    def calc_hei(n):
        now=max(0,int(n**0.5)-1)
        while now**2<n:
            now+=1
        return now**2
    hei=set(i**2 for i in range(200))
    for i in range(1,n+1):
        for j in range(i+1,n+1):
            if i==1:
                now=base+ws[2+size*2+(j-2)]
                ans=[0]
            else:
                now=base+ws[2+size*2+(i-2)]+ws[2+size*2+(j-2)]
                ans=[2+size*2+(i-2),0]
            dif=calc_hei(now)-now
            now=2
            for _ in range(size):
                if dif%2==0:
                    ans.append(now)
                else:
                    ans.append(now+1)
                now+=2
                dif//=2
            ans.append(1)
            ans.append(2+2*size+(j-2))
            print(len(ans),*[_+1 for _ in ans])#,i,j,moku)

if __name__=="__main__":
    main()
0