結果

問題 No.1103 Directed Length Sum
ユーザー akun0716akun0716
提出日時 2020-07-06 20:58:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 676 ms / 3,000 ms
コード長 1,844 bytes
コンパイル時間 943 ms
コンパイル使用メモリ 87,516 KB
実行使用メモリ 136,352 KB
最終ジャッジ日時 2024-09-24 18:10:50
合計ジャッジ時間 7,477 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
28,100 KB
testcase_01 AC 8 ms
30,020 KB
testcase_02 AC 321 ms
136,352 KB
testcase_03 AC 176 ms
50,380 KB
testcase_04 AC 330 ms
50,788 KB
testcase_05 AC 676 ms
60,680 KB
testcase_06 AC 177 ms
42,180 KB
testcase_07 AC 41 ms
30,280 KB
testcase_08 AC 60 ms
35,400 KB
testcase_09 AC 26 ms
29,508 KB
testcase_10 AC 80 ms
35,084 KB
testcase_11 AC 394 ms
51,640 KB
testcase_12 AC 183 ms
41,948 KB
testcase_13 AC 85 ms
35,140 KB
testcase_14 AC 23 ms
29,120 KB
testcase_15 AC 130 ms
37,852 KB
testcase_16 AC 443 ms
52,636 KB
testcase_17 AC 478 ms
53,964 KB
testcase_18 AC 78 ms
34,932 KB
testcase_19 AC 392 ms
51,024 KB
testcase_20 AC 31 ms
31,940 KB
testcase_21 AC 52 ms
31,304 KB
testcase_22 AC 290 ms
45,896 KB
testcase_23 AC 152 ms
40,304 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:78:12: warning: 'root' may be used uninitialized [-Wmaybe-uninitialized]
   78 |         dfs(root, -1);
      |         ~~~^~~~~~~~~~
main.cpp:74:13: note: 'root' was declared here
   74 |         int root;
      |             ^~~~

ソースコード

diff # 

#include <iostream>
#include<vector>
#include<algorithm>
#include<string>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<math.h>
using namespace std;
typedef long long ll;
#define int long long
#define double long double
typedef vector<int> VI;
typedef pair<int, int> pii;
typedef vector<pii> VP;
typedef vector<string> VS;
typedef priority_queue<int> PQ;
template<class T>bool chmax(T &a, const T &b) { if (a < b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b < a) { a = b; return 1; } return 0; }
#define fore(i,a) for(auto &i:a)
#define REP(i,n) for(int i=0;i<n;i++)
#define eREP(i,n) for(int i=0;i<=n;i++)
#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define eFOR(i,a,b) for(int i=(a);i<=(b);++i)
#define SORT(c) sort((c).begin(),(c).end())
#define rSORT(c) sort((c).rbegin(),(c).rend())
#define LB(x,a) lower_bound((x).begin(),(x).end(),(a))
#define UB(x,a) upper_bound((x).begin(),(x).end(),(a))
#define INF 1000000000
#define LLINF 9223372036854775807
#define mod 1000000007
#define eps 1e-12 
//priority_queue<int,vector<int>, greater<int> > q2;

int N;
int ans[1000010];
VI G[1000010];
int t[1000010];//size of subtree

void dfs(int now, int pa) {

	int res = 0;
	fore(to, G[now]) {
		dfs(to, now);
		res += t[to] + ans[to];
		res %= mod;
	}
	ans[now] = res;
}
void dfs1(int now, int pa) {
	fore(to, G[now]) {
		if (to == pa)continue;
		dfs1(to, now);
	}
	t[now]++;
	if (pa != -1)t[pa] += t[now];
	return;
}


signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);

	cin >> N;
	vector<bool>F(N, false);
	REP(i, N - 1) {
		int a, b; cin >> a >> b;
		a--; b--;
		G[a].push_back(b);
		F[b] = 1;
	}
	int root;
	REP(i, N)if (!F[i])root = i;
	
	dfs1(root, -1);
	dfs(root, -1);
	int sum = 0;
	REP(i, N) {
		sum += ans[i];
		sum %= mod;
	}
	cout << sum << endl;

	return 0;
}
0