結果

問題 No.1124 Earthquake Safety
ユーザー pockynypockyny
提出日時 2024-10-20 15:40:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 857 ms / 3,000 ms
コード長 2,516 bytes
コンパイル時間 1,245 ms
コンパイル使用メモリ 85,148 KB
実行使用メモリ 221,472 KB
最終ジャッジ日時 2024-10-20 15:40:35
合計ジャッジ時間 28,111 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
14,080 KB
testcase_01 AC 5 ms
14,076 KB
testcase_02 AC 5 ms
14,124 KB
testcase_03 AC 5 ms
14,728 KB
testcase_04 AC 4 ms
14,504 KB
testcase_05 AC 5 ms
14,040 KB
testcase_06 AC 6 ms
14,684 KB
testcase_07 AC 18 ms
15,344 KB
testcase_08 AC 156 ms
18,012 KB
testcase_09 AC 563 ms
25,044 KB
testcase_10 AC 708 ms
221,472 KB
testcase_11 AC 5 ms
13,992 KB
testcase_12 AC 6 ms
15,004 KB
testcase_13 AC 5 ms
14,156 KB
testcase_14 AC 595 ms
24,908 KB
testcase_15 AC 607 ms
24,820 KB
testcase_16 AC 616 ms
24,868 KB
testcase_17 AC 604 ms
25,056 KB
testcase_18 AC 607 ms
24,912 KB
testcase_19 AC 644 ms
24,968 KB
testcase_20 AC 627 ms
24,924 KB
testcase_21 AC 662 ms
24,892 KB
testcase_22 AC 631 ms
24,836 KB
testcase_23 AC 638 ms
24,824 KB
testcase_24 AC 588 ms
24,860 KB
testcase_25 AC 617 ms
25,852 KB
testcase_26 AC 594 ms
25,652 KB
testcase_27 AC 618 ms
36,116 KB
testcase_28 AC 652 ms
36,352 KB
testcase_29 AC 741 ms
71,276 KB
testcase_30 AC 719 ms
71,524 KB
testcase_31 AC 851 ms
146,116 KB
testcase_32 AC 847 ms
132,068 KB
testcase_33 AC 815 ms
126,100 KB
testcase_34 AC 857 ms
142,724 KB
testcase_35 AC 561 ms
24,516 KB
testcase_36 AC 583 ms
24,532 KB
testcase_37 AC 570 ms
25,636 KB
testcase_38 AC 599 ms
25,544 KB
testcase_39 AC 566 ms
25,228 KB
testcase_40 AC 563 ms
26,008 KB
testcase_41 AC 581 ms
25,532 KB
testcase_42 AC 525 ms
25,888 KB
testcase_43 AC 557 ms
26,184 KB
testcase_44 AC 518 ms
26,240 KB
testcase_45 AC 534 ms
25,980 KB
testcase_46 AC 553 ms
25,940 KB
testcase_47 AC 525 ms
25,836 KB
testcase_48 AC 543 ms
25,592 KB
testcase_49 AC 523 ms
25,568 KB
testcase_50 AC 6 ms
14,312 KB
testcase_51 AC 5 ms
14,096 KB
testcase_52 AC 5 ms
14,044 KB
testcase_53 AC 5 ms
14,096 KB
testcase_54 AC 5 ms
14,068 KB
testcase_55 AC 5 ms
13,972 KB
testcase_56 AC 6 ms
14,780 KB
testcase_57 AC 5 ms
14,976 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <atcoder/modint>
#include <vector>

using namespace std;
using namespace atcoder;
using mint = modint1000000007;
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";
}
0