結果

問題 No.1373 Directed Operations
コンテスト
ユーザー FromBooska
提出日時 2023-02-23 08:31:25
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,312 bytes
コンパイル時間 255 ms
コンパイル使用メモリ 82,500 KB
実行使用メモリ 97,136 KB
最終ジャッジ日時 2024-07-23 06:14:18
合計ジャッジ時間 5,002 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other TLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

# N-1の長さは1本までしか使えない、N-2の長さは2本までしか使えない、、、、
# 1本でも減れば木にならないのでアウト
# そして、gcd(A)=1でなければつながらない
# 構築はUnion Findで愚直につながっていないところをつなげるか、時間間に合うか

class UnionFind():
    def __init__(self, n):
        self.n = n
        self.parents = [-1] * n

    def find(self, x):
        if self.parents[x] < 0:
            return x
        else:
            self.parents[x] = self.find(self.parents[x])
            return self.parents[x]

    def unite(self, x, y):
        x = self.find(x)
        y = self.find(y)
        if x == y:
            return
        if self.parents[x] > self.parents[y]:
            x, y = y, x
        self.parents[x] += self.parents[y]
        self.parents[y] = x

    def size(self, x):
        return -self.parents[self.find(x)]

    def same(self, x, y):
        return self.find(x) == self.find(y)

    def members(self, x):
        root = self.find(x)
        return [i for i in range(self.n) if self.find(i) == root]

    def roots(self):
        return [i for i, x in enumerate(self.parents) if x < 0]

    def group_count(self):
        return len(self.roots())

    def all_group_members(self):
        group_members = defaultdict(list)
        for member in range(self.n):
            group_members[self.find(member)].append(member)
        return group_members

    def __str__(self):
        return '\n'.join(f'{r}: {m}' for r, m in self.all_group_members().items())


N = int(input())
A = list(map(int, input().split()))
from collections import Counter
counted = Counter(A)

length_test = True
for d in range(N-1, 0, -1):
    if counted[d] > N-d:
        length_test = False

import math
import functools
GCD = functools.reduce(math.gcd, A)
if GCD == 1:
    GCD_test = True
else:
    GCD_test = False
    
def make_connection():
    UF = UnionFind(N+1)
    ans_list = []
    for a in A:
        for i in range(1, N+1):
            if UF.same(i, i+a) == False:
                UF.unite(i, i+a)
                ans_list.append(i)
                break
    for a in ans_list:
        print(a)

if length_test == True and GCD_test == True:
    print('YES')
    make_connection()   
else:
    print('NO')
0