結果

問題 No.1103 Directed Length Sum
ユーザー bonKotsubonKotsu
提出日時 2021-05-31 14:03:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,259 ms / 3,000 ms
コード長 869 bytes
コンパイル時間 1,966 ms
コンパイル使用メモリ 173,076 KB
実行使用メモリ 112,896 KB
最終ジャッジ日時 2024-11-08 21:16:03
合計ジャッジ時間 14,951 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 20 ms
26,880 KB
testcase_01 AC 20 ms
26,880 KB
testcase_02 AC 760 ms
112,896 KB
testcase_03 AC 505 ms
38,620 KB
testcase_04 AC 678 ms
40,192 KB
testcase_05 AC 1,259 ms
50,176 KB
testcase_06 AC 421 ms
35,456 KB
testcase_07 AC 92 ms
28,800 KB
testcase_08 AC 134 ms
29,952 KB
testcase_09 AC 63 ms
28,160 KB
testcase_10 AC 191 ms
31,232 KB
testcase_11 AC 755 ms
41,472 KB
testcase_12 AC 427 ms
35,584 KB
testcase_13 AC 199 ms
31,232 KB
testcase_14 AC 53 ms
28,032 KB
testcase_15 AC 330 ms
33,664 KB
testcase_16 AC 848 ms
43,264 KB
testcase_17 AC 894 ms
44,032 KB
testcase_18 AC 195 ms
31,104 KB
testcase_19 AC 760 ms
41,856 KB
testcase_20 AC 73 ms
28,336 KB
testcase_21 AC 121 ms
29,568 KB
testcase_22 AC 610 ms
39,168 KB
testcase_23 AC 353 ms
34,048 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:50:10: warning: 'r' may be used uninitialized [-Wmaybe-uninitialized]
   50 |   dfs_sum(r,0);
      |   ~~~~~~~^~~~~
main.cpp:41:7: note: 'r' was declared here
   41 |   int r;
      |       ^

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define rep(i, n) for (int i = 0; i < (n); i++)
#define P pair<int, int>

const int MOD = 1e9+7;

vector<int> G[1000005];
int sum[1000005];
vector<int> dist;

void dfs(int v, int d) {
  dist[v] = d;
  for (int nv : G[v]) {
    dfs(nv, d+1);
  }
}

void dfs_sum(int v, int d) {
  sum[v] = d + dist[v];
  sum[v] %= MOD;
  for (int nv : G[v]) {
    dfs_sum(nv, sum[v]);
  }
}

int main() {
  int n;
  cin >> n;
  vector<bool> seen(n, true);
  rep(i,n-1) {
    int a, b;
    cin >> a >> b;
    a--, b--;
    G[a].push_back(b);
    seen[b] = false;
  }

  int r;
  // 根を決定
  rep(i,n) {
    if (seen[i]) r = i;
  }

  const int INF = 1001001001;
  dist.resize(n, INF);
  dfs(r, 0);
  dfs_sum(r,0);

  int ans = 0;
  rep(i,n) {
    ans += sum[i];
    ans %= MOD;
  }
  cout << ans << endl;
  






}
0