結果

問題 No.1103 Directed Length Sum
ユーザー kacho65535kacho65535
提出日時 2020-12-23 22:29:44
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,523 ms / 3,000 ms
コード長 1,635 bytes
コンパイル時間 2,212 ms
コンパイル使用メモリ 176,404 KB
実行使用メモリ 190,768 KB
最終ジャッジ日時 2024-09-21 16:35:11
合計ジャッジ時間 17,521 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 830 ms
190,768 KB
testcase_03 AC 542 ms
97,424 KB
testcase_04 AC 768 ms
58,432 KB
testcase_05 AC 1,523 ms
98,560 KB
testcase_06 AC 475 ms
39,032 KB
testcase_07 AC 81 ms
12,032 KB
testcase_08 AC 132 ms
16,456 KB
testcase_09 AC 50 ms
8,832 KB
testcase_10 AC 197 ms
21,376 KB
testcase_11 AC 845 ms
63,360 KB
testcase_12 AC 480 ms
39,404 KB
testcase_13 AC 201 ms
21,744 KB
testcase_14 AC 36 ms
7,296 KB
testcase_15 AC 346 ms
31,360 KB
testcase_16 AC 905 ms
70,712 KB
testcase_17 AC 1,103 ms
73,728 KB
testcase_18 AC 199 ms
21,352 KB
testcase_19 AC 844 ms
64,784 KB
testcase_20 AC 60 ms
9,984 KB
testcase_21 AC 116 ms
15,232 KB
testcase_22 AC 705 ms
53,504 KB
testcase_23 AC 372 ms
33,408 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define mod 1000000007ll
#define loop(i, n) for (int i = 0; i < n; i++)
#define all(v) v.begin(), v.end()
#define putout(a) cout << a << '\n'
#define Sum(v) accumulate(all(v), 0ll)
void bfs(vector<vector<ll>> &G, vector<ll> &dist, ll s)
{
    queue<ll> que;
    dist[s] = 0;
    que.push(s);
    while (!que.empty())
    {
        ll v = que.front();
        que.pop();
        for (auto next_v : G[v])
        {
            if (dist[next_v] != -1)
                continue;
            dist[next_v] = dist[v] + 1;
            que.push(next_v);
        }
    }
}
vector<ll> sz;
void dfs(vector<vector<ll>> &G, ll v, ll p)
{
    for (auto nv : G[v])
    {
        if (nv == p)
            continue;
        dfs(G, nv, v);
    }
    sz[v] = 1;
    for (auto nv : G[v])
    {
        if (nv == p)
            continue;
        sz[v] += sz[nv];
    }
}
int main()
{
    ll N;
    cin >> N;
    vector<vector<ll>> G(N);
    ll root = 0; //bとして1度も出現しないのが根
    sz.resize(N);
    vector<ll> A(N - 1), B(N - 1), x(N - 1), y(N - 1), used(N, false);
    loop(i, N - 1)
    {
        cin >> A[i] >> B[i];
        A[i]--;
        B[i]--;
        used[B[i]] = true;
        G[A[i]].emplace_back(B[i]);
    }
    loop(i, N) if (!used[i]) root = i;
    vector<ll> dist(N, -1);
    bfs(G, dist, root);
    dfs(G, root, -1);
    loop(i, N - 1) x[i] = dist[B[i]]; //dist[A[i]]+1でもよい
    loop(i, N - 1) y[i] = sz[B[i]];
    ll ans = 0;
    loop(i, N - 1)
    {
        ans += x[i] * y[i] % mod;
        ans %= mod;
    }
    putout(ans);
    return 0;
}
0