結果

問題 No.1103 Directed Length Sum
ユーザー snrnsidy
提出日時 2021-10-23 17:44:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 899 ms / 3,000 ms
コード長 1,046 bytes
コンパイル時間 2,695 ms
コンパイル使用メモリ 194,404 KB
最終ジャッジ日時 2025-01-25 04:56:33
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

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