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;  }