2435: [Noi2011]道路修建
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 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;
}