【解题报告】CodeVS3728 联合权值

先看原题：

题目描述 Description

输入描述 Input Description

输出描述 Output Description

样例输入 Sample Input

样例输出 Sample Output

数据范围及提示 Data Size & Hint

思路分析：因为这个联通图有N个点和(N-1)条边，所以肯定是一张无环图。我用vector存储每条边，设置了vector<int>t[200010]，这样t[i]中存储的元素就是点i所连接的点。依次读入后进行处理，我本想用一个for(int i=1;i<=n;i++)遍历所有点，嵌套for(int j=0;j<t[i].size();j++)遍历这个点所连接的各个点，再嵌套for(int k=0;k<t[t[i][j]].size();k++)遍历这个点连接的各个点所连接的各个点，然后累加求sum，比较求max。——很当然的O(n^3)的想法，数据也是很当然的超时了，10个点只拿了7个。后来看了看别人的题解，找到了新的思路：用一个for(int i=1;i<=n;i++)遍历所有点，然后嵌套一个for(int j=0;j<t[i].size();j++)遍历这个点所连接的各个点——到这里还和我一样，但是——它并没有再来一层循环，它只是记录下了在第二层循环中的点的权值的最大值和次大值，将其乘积与max比较大小；同时记录下了第二层循环中的所有点的权值之和，在第二层循环出来后又放上一个for(int j=0;j<t[i].size();j++)的循环，然后用sum=(sum+t[i][j]的权值*(第二层循环中的总权值-t[i][j]的权值))%10007来求得总联合权值。这个算法很巧妙地将时间复杂度从O(n^3)降到了O(n^2)。

求一组数据两两之间的乘积之和的两种算法对比：

常规(O(n^2))：

for(int i=1;i<n;i++)

for(int j=1;j<n;j++)

if(i==j)continue;

else sum+=a[i]*a[j];

高效(O(n)):

for(int i=1;i<n;i++)

tmps+=a[i];

for(int i=1;i<n;i++)

sum+=a[i]*(tmps-a[i]);

一个小插曲：

以前一直以为vector的获取元素个数的函数是.capacity()……结果为此查了半天的错……   .size()才是获取元素个数的函数，.capacity()只是获取允许存储的元素个数而已。

最后，上代码：

#include<cstdio>
#include<vector>
using namespace std;
int wei[200010];
vector<int> t[200010];
int main(){
	int n,sum=0,max=0,t1,t2;
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		scanf("%d %d",&t1,&t2);
		t[t1].push_back(t2);
		t[t2].push_back(t1);
	}
	for(int i=1;i<=n;i++)
		scanf("%d",&wei[i]);
	for(int i=1;i<=n;i++){
		int max1=0,max2=0,tmps=0;
		for(int j=0;j<t[i].size();j++){
			int tmpw=wei[t[i][j]];
			if(tmpw>max1){max2=max1;max1=tmpw;}
			else if(tmpw>max2)max2=tmpw;
			tmps=(tmps+tmpw)%10007;
		}
		int tmp=max1*max2;
		max=max<tmp?tmp:max;
		for(int j=0;j<t[i].size();j++){
			int tmpw=wei[t[i][j]];
			sum=(sum+tmpw*(tmps-tmpw))%10007;
		}
	}
	printf("%d %d\n",max,sum);
	return 0;
}