P2052 [NOI2011]道路修建 图变树 dfs【1】

题目描述

在 W 星球上有 n 个国家。为了各自国家的经济发展，他们决定在各个国家 之间建设双向道路使得国家之间连通。但是每个国家的国王都很吝啬，他们只愿 意修建恰好 n – 1 条双向道路。 每条道路的修建都要付出一定的费用，这个费用等于道路长度乘以道路两端 的国家个数之差的绝对值。例如，在下图中，虚线所示道路两端分别有 2 个、4 个国家，如果该道路长度为 1，则费用为 1×|2 – 4|=2。图中圆圈里的数字表示国 家的编号。 

由于国家的数量十分庞大，道路的建造方案有很多种，同时每种方案的修建 费用难以用人工计算，国王们决定找人设计一个软件，对于给定的建造方案，计 算出所需要的费用。请你帮助国王们设计一个这样的软件。

输入输出格式

输入格式：

输入的第一行包含一个整数 n，表示 W 星球上的国家的数量，国家从 1 到 n 编号。 接下来 n – 1 行描述道路建设情况，其中第 i 行包含三个整数 ai、bi和 ci，表 示第 i 条双向道路修建在 ai与 bi两个国家之间，长度为 ci。

输出格式：

输出一个整数，表示修建所有道路所需要的总费用。

输入输出样例

输入样例#1： 复制

6
1 2 1
1 3 1
1 4 2
6 3 1
5 2 1

输出样例#1： 复制

20

第一次，原原本本按照题意，代码40分

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
int n,a,b,c;
vector< vector<int> > p,cost,num;
vector<int> sumnum;
long long res=0,res1;
int  todp(int k1,int k2)
{

	//if(num[k1][k2]>0) return num[k1][k2];
	int realnum2=p[k1][k2];
	int sum=1;
	for(int i=0;i<p[realnum2].size();i++)
	{
		if(p[realnum2][i]!=k1)
		sum+=todp(realnum2,i);
	}
	return sum;
}
int main()
{
    cin>>n;
    p.resize(n+1); 
    cost.resize(n+1); 
    num.resize(n+1); 
    sumnum.resize(n+1);
    for(int i=0;i<n-1;i++)
    {
    	scanf("%d%d%d",&a,&b,&c);
    	p[a].push_back(b);
    	cost[a].push_back(c);
    	num[a].push_back(0);
    	
    	p[b].push_back(a);
    	cost[b].push_back(c);
    	num[b].push_back(0);
    }
    for(int i=1;i<=n;i++)
    {
    	for(int j=0;j<p[i].size();j++)
    	{
	    	num[i][j]=todp(i,j);
	    }
    }
    for(int i=1;i<=n;i++)
    {
    	for(int j=0;j<p[i].size();j++)
    	{
    		int realnum2=p[i][j],index;
    		for(int k=0;k<p[realnum2].size();k++)
    		{
		    	if(p[realnum2][k]==i) 
		    	{
	    			index=k;
	    			break;
	    		}
		    }
	    	res1=(long long)cost[i][j]*
			abs(num[i][j]-num[realnum2][index]);
			res+=res1;
	    }
    }
    cout<<res/2<<endl;
    return 0;
}

第二次，修改了算法。75分  题目中说到n个点n-1条边那么自然想到这是一棵树咯，一颗无根树，我们可以给它随意定义一个根，这里定义1号为根，然后判断每个节点的子树有多少个节点，然后就知道这条边的贡献为abs(n-2*size_v)*w[i]

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
#include<map>
using namespace std;
int n,a,b,c;
vector< vector<int> > p,cost;
vector<int> sumnum,fa,num;
long long res=0,res1;
inline  int dfs(int r) {
	int sum=1;
	if(num[r]>0) return num[r];
	for(int i=0; i<p[r].size(); i++) {
		if(fa[p[r][i]]==0) {
			fa[p[r][i]]=r;
			sum+=dfs(p[r][i]);
		}
	}
	num[r]=sum;
	return sum;
}
int main() {
	cin>>n;
	p.resize(n+1);
	cost.resize(n+1);
	num.resize(n+1);
	fa.resize(n+1);
	for(int i=0; i<n-1; i++) {
		scanf("%d%d%d",&a,&b,&c);
		p[a].push_back(b);
		cost[a].push_back(c);

		p[b].push_back(a);
		cost[b].push_back(c);
	}
	int root;
	for(int i=1; i<=n; i++) {
		if(p[i].size()==1) {
			root=i;
			break;
		}
	}
	fa[root]=root;
	num[root]=1;
	for(int i=0; i<p[root].size(); i++) {
		fa[p[root][i]]=root;
		num[p[root][i]]=dfs(p[root][i]);
		num[root]+=num[p[root][i]];
	}// tree
	for(int i=1; i<=n; i++) {
		for(int j=0; j<p[i].size(); j++) {
			int p1=i;
			int p2=p[i][j];
			if(fa[p1]==p2) {
				res1=(long long)cost[i][j]*
				     abs(num[p1]-(n-num[p1]));
				res+=res1;
			}
		}
	}
	cout<<res<<endl;
	return 0;
}

但是开了O(2）优化，100分