BZOJ 2435（BFS+输入输出外挂）

2435: [Noi2011]道路修建

Time Limit: 10 Sec Memory Limit: 128 MB

Submit: 2299 Solved: 682

[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

算法描述：此题乍一看DFS;实际上 从数据量看100W的数据 DFS肯定会爆栈;因为DFS一般是函数递归形式调用;每次函数递归都要进栈;数据量大了，就会爆栈；

所以BFS解决！ 不过我在写的时候直接上BFS 也WR了！后来看大神的解题代码！发现加上输入输出外挂后立马 AC！

/**********************************************

本题必须用到输入输出外挂

****************************************************/

#include <iostream>

#include <cstdio>

#include <cstring>

#include <queue>

using namespace std;

#define N 1001000

typedef long long LL;

struct Node {

int value,fa,s,next;

Node(){}

Node(int _v,int _next,int _fa,int _s){

value = _v; next = _next;

fa = _fa; s = _s;

}

}node[N<<1],stack[N<<1];

queue<Node> q;

LL Abs(LL a){ return a >= 0? a : -a; }

int head[N<<1],top,n,siz
;

void add_Node(int a,int b,int value) {

node[top].fa = a;

node[top].s = b;

node[top].value = value;

node[top].next = head[a];

head[a] = top++;

}

void bfs(int root,LL &ans) {

while(!q.empty()) q.pop();

int top_stack = 0;

q.push(Node(0,-1,-1,root));

while(!q.empty()) {

Node cur = q.front(); q.pop();

stack[top_stack++] = Node(cur.value,cur.next,cur.fa,cur.s);

for(int i = head[cur.s];i!=-1;i = node[i].next){

if(cur.fa == node[i].s) continue;

q.push(Node(node[i].value,node[i].next,node[i].fa,node[i].s));

}

}

for(int i = top_stack-1;i>=0;i--){

if(stack[i].fa != -1) siz[stack[i].fa] += siz[stack[i].s];

ans += stack[i].value*Abs(2*siz[stack[i].s] - n);

}

}

void init(){

memset(head,-1,sizeof(head));

for(int i = 1;i<=n;i++) siz[i] = 1;

top = 0;

}

inline LL get(){

LL ret = 0;

char c = getchar();

while(!isdigit(c)) c = getchar();

while(isdigit(c)){

ret = ret*10+(LL)(c - '0');

c = getchar();

}

return ret;

}

inline void out(LL x){

if(x > 9) out(x/10);

putchar(x%10+'0');

}

int main(){

int a,b,c;

//freopen("road20.in","r",stdin);

n = get();

init();

for(int i = 1;i<n;i++){

a = get(); b = get(); c = get();

add_Node(a,b,c);

add_Node(b,a,c);

}

LL ans = 0;

bfs(1,ans);

out(ans);

return 0;

}