結果

問題 No.1103 Directed Length Sum
コンテスト
ユーザー bonKotsu
提出日時 2021-05-31 14:03:30
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
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