【求助零水】P1351求助
查看原帖
【求助零水】P1351求助
399150
Shunpower楼主2021/8/15 19:50

RT

思路类似 wucstdio 的题解。

#include <bits/stdc++.h>
const int m=10007;
using namespace std;
int n;
vector <int> p[200010];
int w[2000010];
int maxx,maxn,sum1,sum2;
int ans=INT_MAX;
int sum;
void dfs(int x,int fa){
	for(int i=0;i<p[x].size();i++){
		int y=p[x][i];
		if(y!=fa){
			dfs(y,x);
		}
		if(w[y]>maxx){
			maxn=maxx;
			maxx=w[y];
		}
		else if(w[y]>maxn){
			maxn=w[y];
		}
		sum1+=w[y];
		sum1%=m;
		sum2+=w[y]*w[y];
		sum2%=m;
	}
	sum1*=sum1;
	sum1%=m;
	sum+=(sum1-sum2+m)%m;
	ans=min(ans,maxx*maxn);
}
int main(){
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int x,y;
		cin>>x>>y;
		p[x].push_back(y);
		p[y].push_back(x);
	}
	for(int i=1;i<=n;i++){
		cin>>w[i];
	}
	dfs(1,20040615+20081210-19950501-20191123);
	cout<<ans<<" "<<sum%m<<endl;
	return 0;
}

调了很久,求助。

2021/8/15 19:50
加载中...