結果

問題 No.1103 Directed Length Sum
ユーザー snrnsidysnrnsidy
提出日時 2021-10-23 17:44:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 761 ms / 3,000 ms
コード長 1,046 bytes
コンパイル時間 2,021 ms
コンパイル使用メモリ 202,868 KB
実行使用メモリ 136,076 KB
最終ジャッジ日時 2024-09-25 08:54:56
合計ジャッジ時間 11,367 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 15 ms
38,848 KB
testcase_01 AC 15 ms
39,492 KB
testcase_02 AC 352 ms
136,076 KB
testcase_03 AC 199 ms
46,200 KB
testcase_04 AC 394 ms
51,588 KB
testcase_05 AC 761 ms
58,312 KB
testcase_06 AC 215 ms
47,556 KB
testcase_07 AC 46 ms
41,928 KB
testcase_08 AC 64 ms
42,688 KB
testcase_09 AC 34 ms
39,880 KB
testcase_10 AC 95 ms
42,944 KB
testcase_11 AC 431 ms
52,296 KB
testcase_12 AC 229 ms
47,172 KB
testcase_13 AC 105 ms
43,204 KB
testcase_14 AC 29 ms
40,900 KB
testcase_15 AC 153 ms
44,740 KB
testcase_16 AC 510 ms
52,932 KB
testcase_17 AC 510 ms
54,132 KB
testcase_18 AC 93 ms
43,076 KB
testcase_19 AC 438 ms
51,524 KB
testcase_20 AC 38 ms
41,924 KB
testcase_21 AC 62 ms
42,820 KB
testcase_22 AC 350 ms
50,244 KB
testcase_23 AC 187 ms
46,792 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
 
using namespace std;
 
vector <int> adj[1000001];
int indegree[1000001];
int n,a,b;
int cnt[1000001];
long long int dp[1000001];
const long long int MOD = 1e9 + 7;

int dfs1(int now)
{
	if(cnt[now]!=-1)
	{
		return cnt[now];
	}
	cnt[now] = 1;
	for(auto next : adj[now])
	{
		cnt[now] += dfs1(next);
	}
	return cnt[now];
}

long long int dfs2(int now)
{
	if(dp[now]!=-1)
	{
		return dp[now];
	}

	dp[now] = 0;
	for(auto next : adj[now])
	{
		dp[now] += (dfs2(next) + cnt[next]);
	}
	return dp[now];
}

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

	cin >> n;

	for(int i=0;i<n-1;i++)
	{
		cin >> a >> b;
		adj[a].push_back(b);
		indegree[b]++;
	}

	int root = -1;
	for(int i=1;i<=n;i++)
	{
		if(indegree[i]==0)
		{
			root = i;
			break;
		}
	}


	memset(cnt,-1,sizeof(cnt));
	memset(dp,-1,sizeof(dp));
	dfs1(root);
	dfs2(root);

	long long int res = 0;

	for(int i=1;i<=n;i++)
	{
		//cout << i << ' ' << cnt[i] << ' ' << dp[i] << '\n';
		res += dp[i];
		res%=MOD;
	}

	cout << res << '\n';

	return 0;	
}
0