結果

問題 No.1124 Earthquake Safety
ユーザー pockynypockyny
提出日時 2024-10-20 15:39:17
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,515 bytes
コンパイル時間 1,138 ms
コンパイル使用メモリ 85,688 KB
実行使用メモリ 221,436 KB
最終ジャッジ日時 2024-10-20 15:39:49
合計ジャッジ時間 31,253 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
13,976 KB
testcase_01 AC 4 ms
14,216 KB
testcase_02 AC 5 ms
13,924 KB
testcase_03 AC 5 ms
14,168 KB
testcase_04 AC 5 ms
14,512 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 6 ms
14,020 KB
testcase_12 AC 5 ms
14,028 KB
testcase_13 AC 4 ms
13,848 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 AC 5 ms
14,048 KB
testcase_51 AC 5 ms
14,804 KB
testcase_52 AC 5 ms
14,248 KB
testcase_53 AC 5 ms
14,040 KB
testcase_54 AC 5 ms
14,284 KB
testcase_55 AC 5 ms
13,960 KB
testcase_56 AC 5 ms
14,808 KB
testcase_57 AC 6 ms
14,016 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <atcoder/modint>
#include <vector>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
vector<int> G[300010];
mint f[300010],g[300010],h[300010];
int dep[300010];
mint pw(mint a,int x){
mint ret = 1;
while(x){
if(x&1) (ret *= a);
(a *= a); x /= 2;
}
return ret;
}
// (i,j,k)
// (i,j,k)
void dfs(int s,int p){
// cout << "s == " << s << " p == " << p << endl;
// /g使
vector<vector<mint>> dpf(4,vector<mint>(2)),ndpf(4,vector<mint>(2));
int i,j;
dpf[0][0] = 1;
for(int v:G[s]){
if(v==p) continue;
dep[v] = dep[s] + 1;
dfs(v,s);
f[s] += f[v]; g[s] += g[v]; h[s] += h[v];
for(i=0;i<=3;i++){
if(i){
for(j=0;j<2;j++) ndpf[i][j] += dpf[i - 1][j]*h[v];
}
if(i>=2){
for(j=0;j<2;j++) ndpf[i][1] += dpf[i - 2][j]*g[v];
}
for(j=0;j<2;j++) ndpf[i][j] += dpf[i][j];
}
for(i=0;i<=3;i++){
for(j=0;j<2;j++) dpf[i][j] = ndpf[i][j], ndpf[i][j] = 0;
}
}
mint pp = pw(2,dep[s]);
mint inv = (mint)1/pp;
h[s] += inv;
g[s] += 2*dpf[2][0]*pp + 2*dpf[1][0];
f[s] += 6*dpf[3][0]*pw(pp,3) + 3*dpf[3][1]*pw(pp,2) + 3*dpf[2][1]*pp + 6*dpf[2][0]*pw(pp,2);
}
int main(){
int i,n; cin >> n;
for(i=0;i<n - 1;i++){
int a,b; cin >> a >> b; a--; b--;
G[a].push_back(b); G[b].push_back(a);
}
dep[0] = 0;
dfs(0,-1);
// (i,i,i)
mint x = n;
// (i,i,j)
mint y = 0;
for(i=0;i<n;i++){
mint p = pw(2,dep[i]);
mint sum = 0,sum2 = 0;
for(int u:G[i]){
if(u==-1 || dep[u]<dep[i]) continue;
sum += h[u];
sum2 += h[u]*h[u];
}
// cout << (sum*4).val() << " " << (sum2*4).val() << " " << p.val() << endl;
y += ((sum*sum - sum2)/2)*pw(p,2);
y += sum*p;
}
y *= 6;
// (i,j,k)
mint z = f[0];
cout << ((x + y + z)*pw(2,n - 1)).val() << "\n";
// for(i=0;i<n;i++){
// mint x = pw(2,n - 1);
// cout << (f[i]*x).val() << " " << (g[i]*x).val() << " " << (h[i]*x).val() << "\n";
// }
// cout << (x*pw(2,n - 1)).val() << " " << (y*pw(2,n - 1)).val() << " " << (z*pw(2,n - 1)).val() << "\n";
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0