记录编号 590796 评测结果 AAAAAAAAAA
题目名称 [SYOI 2015] Asm.Def的基本算法 最终得分 100
用户昵称 Gravatar袁书杰 是否通过 通过
代码语言 C++ 运行时间 0.140 s
提交时间 2024-07-11 16:47:26 内存使用 6.24 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
long long n,s,m,u,v,w[100005],ans,sum[100005];
vector<long long> a[100005];
void dfs(long long u,long long father) {
    sum[u]=w[u];
	for(long long v=0; v<a[u].size(); v++) {
		if(a[u][v]==father) {
			continue;
		}
		dfs(a[u][v],u);
		ans+=sum[u]%1000000007*sum[a[u][v]]%1000000007*w[u]%1000000007;
		ans%=1000000007;
		sum[u]+=sum[a[u][v]]%1000000007;
		sum[u]%=1000000007;
	}
}
int main() {
	freopen("asm_algo.in","r",stdin);
	freopen("asm_algo.out","w",stdout);
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>w[1];
	s=1;
	for(long long i=2; i<=n; i++) {
		cin>>u>>w[i];
		a[u].push_back(i);
	}
	dfs(1,1);
	ans*=2;
	ans%=1000000007;
	for(long long i=1;i<=n;i++){
	    ans+=w[i]%1000000007*w[i]%1000000007*w[i]%1000000007;
	    ans%=1000000007;
    }
    ans%=1000000007;
    cout<<ans;
	return 0;
}