BZOJ2435 道路修建 [DFS][内嵌汇编]

2435: [Noi2011]道路修建

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 4084   Solved: 1420
[ 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

从PoPoqqq爷那里学了内嵌汇编。。。手动开栈过了这题，还挺快的，BZOJ加强的数据，以前的dfs过不了了。
话说NOI day2 t1 还居然很水。。。。
模拟题?

#include<iostream>  
#include<cstdio>  
#include<cstdlib>  
#include<cstring>
#include<algorithm>
#define N 1000005
#define pairs pair<int,int>
#define fi first
#define se second
#pragma comment(linker, “/STACK:1024000000,1024000000”)
#define stack_size (20001000) 
using namespace std;
int stack[stack_size],bak;
int n,tot,head[N],fa[N],sz[N];
long long ans;  
struct data{
    int to,nxt,len;
}E[N<<1];
void addedge(int u,int v,int len)  {
    E[++tot].to=v;E[tot].len=len;E[tot].nxt=head[u];head[u]=tot;  
    E[++tot].to=u;E[tot].len=len;E[tot].nxt=head[v];head[v]=tot;  
}
void dfs(int x,int f){  
    sz[x]=1;  
    for(register int v,i=head[x];i;i=E[i].nxt){
        if(v=E[i].to,v==f)continue;  
        dfs(v,x);
        sz[x]+=sz[v];  
        ans+=1LL*E[i].len*abs(n-2*sz[v]);
    }
}
void DFS()    
{    
    __asm__ __volatile__    
    (    
        "mov %%esp,%0\n"  
        "mov %1,%%esp\n"  
        :"=g"(bak)    
        :"g"(stack+stack_size-1)    
        :    
    );    
    dfs(1,0);    
    __asm__ __volatile__    
    (    
        "mov %0,%%esp\n"  
        :    
        :"g"(bak)    
        :    
    );    
} 
inline int getc() {  
    static const int L = 1 << 15;  
    static char buf[L], *S = buf, *T = buf;  
    if (S == T) {  
        T = (S = buf) + fread(buf, 1, L, stdin);  
        if (S == T)  
            return EOF;  
    }  
    return *S++;  
}  
inline int getint() {  
    int c;  
    while(!isdigit(c = getc()) && c != '-');  
    bool sign = c == '-';  
    int tmp = sign ? 0 : c - '0';  
    while(isdigit(c = getc()))  
        tmp = (tmp << 1) + (tmp << 3) + c - '0';  
    return sign ? -tmp : tmp;  
}
int main(){
    n=getint();
    for(register int x,y,z,i=1;i<n;i++)
        x=getint(),y=getint(),z=getint(),addedge(x,y,z);  
    DFS();
    printf("%lld",ans);
    return 0;  
}