結果

問題 No.1103 Directed Length Sum
ユーザー onsen_manjuuuonsen_manjuuu
提出日時 2020-07-05 16:19:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,069 ms / 3,000 ms
コード長 816 bytes
コンパイル時間 1,744 ms
コンパイル使用メモリ 175,312 KB
実行使用メモリ 136,068 KB
最終ジャッジ日時 2024-09-22 13:45:30
合計ジャッジ時間 12,049 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 773 ms
136,068 KB
testcase_03 AC 497 ms
50,672 KB
testcase_04 AC 585 ms
35,816 KB
testcase_05 AC 1,069 ms
59,428 KB
testcase_06 AC 367 ms
24,392 KB
testcase_07 AC 71 ms
8,448 KB
testcase_08 AC 110 ms
11,132 KB
testcase_09 AC 43 ms
6,940 KB
testcase_10 AC 159 ms
13,956 KB
testcase_11 AC 658 ms
38,820 KB
testcase_12 AC 370 ms
24,592 KB
testcase_13 AC 163 ms
14,308 KB
testcase_14 AC 33 ms
6,940 KB
testcase_15 AC 287 ms
19,936 KB
testcase_16 AC 747 ms
43,036 KB
testcase_17 AC 780 ms
44,712 KB
testcase_18 AC 161 ms
13,772 KB
testcase_19 AC 674 ms
39,672 KB
testcase_20 AC 56 ms
7,040 KB
testcase_21 AC 101 ms
10,324 KB
testcase_22 AC 545 ms
32,824 KB
testcase_23 AC 302 ms
21,000 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>

using namespace std;
using ll = long long;
const ll mod = 1e9 + 7;


ll n, root;
vector<vector<ll>> hen;
vector<ll> dp;



pair<ll,ll> dfs(ll v, ll par) {
    ll sum = 0;
    ll cnt = 1;

    for(auto i : hen[v]) {
        pair<ll,ll> cur = dfs(i, v);
        sum += (cur.first + cur.second);
        cnt += cur.second;
        sum %= mod;
    }
    dp[v] = sum;
    return {sum, cnt};
}



int main()
{
    cin >> n;
    hen.resize(n), dp.resize(n);
    vector<ll> fin(n);

    for(ll i = 0; i < n - 1; i++) {
        ll a, b; cin >> a >> b; a--, b--;
        hen[a].emplace_back(b);
        fin[b]++;
    }
    root = find(fin.begin(), fin.end(), 0) - fin.begin();


    
    dfs(root, -1);
    cout << accumulate(dp.begin(), dp.end(),0LL, [](ll a, ll b){return (a + b) % mod;}) << endl;
}
0