DFS/DP

　　直接dfs就好了……x->y val=c  :  ans+=abs(n-size[y]-size[y])*c;

　　然而为啥会一直RE呢？

　　后来改成dfs返回当前节点的size就过了……过了……

　　100W的节点难道linux下也会爆栈嘛= =？sad

 1 /**************************************************************
 2     Problem: 2435
 3     User: Tunix
 4     Language: C++
 5     Result: Accepted
 6     Time:5292 ms
 7     Memory:73348 kb
 8 ****************************************************************/
 9
10 //BZOJ 2435
11 #include<vector>
12 #include<cstdio>
13 #include<cstring>
14 #include<cstdlib>
15 #include<iostream>
16 #include<algorithm>
17 #define rep(i,n) for(int i=0;i<n;++i)
18 #define F(i,j,n) for(int i=j;i<=n;++i)
19 #define D(i,j,n) for(int i=j;i>=n;--i)
20 #define pb push_back
21 using namespace std;
22 inline int getint(){
23     int v=0,sign=1; char ch=getchar();
24     while(ch<‘0‘||ch>‘9‘){ if (ch==‘-‘) sign=-1; ch=getchar();}
25     while(ch>=‘0‘&&ch<=‘9‘){ v=v*10+ch-‘0‘; ch=getchar();}
26     return v*sign;
27 }
28 const int N=1001000,INF=~0u>>2;
29 typedef long long LL;
30 /******************tamplate*********************/
31
32 int to[N<<1],nxt[N<<1],head[N],cnt,l[N<<1];
33 void add(int x,int y,int z){
34     to[++cnt]=y; nxt[cnt]=head[x]; head[x]=cnt; l[cnt]=z;
35     to[++cnt]=x; nxt[cnt]=head[y]; head[y]=cnt; l[cnt]=z;
36 }
37 int n,s[N];
38 LL ans;
39
40 int dfs(int x,int fa){
41     s[x]=1;
42     for(int i=head[x];i;i=nxt[i])
43         if (to[i]!=fa){
44             s[x]+=dfs(to[i],x);
45             ans+=(LL)abs(n-s[to[i]]*2)*l[i];
46         }
47     return s[x];
48 }
49 int main(){
50 #ifndef ONLINE_JUDGE
51     freopen("2435.in","r",stdin);
52     freopen("2435.out","w",stdout);
53 #endif
54     n=getint();
55     F(i,2,n){
56         int x=getint(),y=getint(),z=getint();
57         add(x,y,z);
58     }
59     dfs(1,0);
60     cout <<ans<<endl;
61     return 0;
62 }

2435: [Noi2011]道路修建

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 2477  Solved: 753
[Submit][Status][Discuss]

Description

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

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

Input

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

Output

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

Sample Input

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

Sample Output

20

HINT

n = 1,000,000 1≤ai, bi≤n

0 ≤ci≤ 10^6

Source

Day2

[Submit][Status][Discuss]