結果

問題 No.2494 Sum within Components
ユーザー timitimi
提出日時 2023-10-06 22:43:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 392 ms / 2,000 ms
コード長 838 bytes
コンパイル時間 510 ms
コンパイル使用メモリ 82,252 KB
実行使用メモリ 160,432 KB
最終ジャッジ日時 2024-07-26 16:45:12
合計ジャッジ時間 3,813 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
53,476 KB
testcase_01 AC 35 ms
53,376 KB
testcase_02 AC 35 ms
52,320 KB
testcase_03 AC 35 ms
52,660 KB
testcase_04 AC 35 ms
53,012 KB
testcase_05 AC 35 ms
52,856 KB
testcase_06 AC 38 ms
54,168 KB
testcase_07 AC 39 ms
54,348 KB
testcase_08 AC 40 ms
52,504 KB
testcase_09 AC 106 ms
77,504 KB
testcase_10 AC 138 ms
79,224 KB
testcase_11 AC 96 ms
77,596 KB
testcase_12 AC 163 ms
79,632 KB
testcase_13 AC 143 ms
78,308 KB
testcase_14 AC 322 ms
103,740 KB
testcase_15 AC 286 ms
99,144 KB
testcase_16 AC 188 ms
101,044 KB
testcase_17 AC 169 ms
160,432 KB
testcase_18 AC 227 ms
148,508 KB
testcase_19 AC 392 ms
127,356 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

N,M=map(int, input().split())
A=list(map(int, input().split()))
B=[0]*N;C=[]
par=[i for i in range(N)]
rank=[0]*(N)
friend=[0]*N
block=[0]*N
size=[1]*N
def find(x):
  if par[x]==x:
    return x
  else:
    par[x]=find(par[x])
    return par[x]
 
#同じ集合か判定
def same(x,y):
  return find(x)==find(y)
 
def union(x,y):
  x=find(x)
  y=find(y)
  if x==y:
    return 
  if rank[x]>rank[y]:
    par[y]=x
    size[x]+=size[y]
  else:
    par[x]=y
    size[y]+=size[x]
    if rank[x]==rank[y]:
      rank[y]+=1
 
 
for i in range(M):
  u,v=map(int, input().split())
  u-=1;v-=1
  C.append((u,v))
  union(u,v)
  
D={}
for i in range(N):
  d=find(i)
  if d not in D:
    D[d]=[]
  D[d].append(i)

ans=1;mod=998244353
for d in D:
  c=0
  for a in D[d]:
    c+=A[a]
  for i in range(len(D[d])):
    ans*=c 
    ans%=mod 
print(ans)
  
  
0