COJ 0252 HDNOIP201304阻断传染

HDNOIP201304阻断传染

难度级别： A； 编程语言：不限；运行时间限制：1000ms； 运行空间限制：51200KB； 代码长度限制：2000000B

试题描述

   H国有n个城市,n个城市用n-1条双向道路相互连通构成一棵树,1号城市是首都，也是树中的根节点。H国的首都爆发了一种危害性极高的传染病。当局为了控制疫情,不让疫情扩散到边境城市(叶子节点所表示的城市)，决定在一些公路上设检疫关卡，但在不同道路上设卡的代价可能不同，为了防止边境城市受传染，必需让从首都到每个边境城市的路径上都至少有一个检疫关卡，请你编程求能阻止传染的最小设卡总代价。

输入

第一行一个整数 n,表示城市个数。
    接下来的 n-1 行,每行 3 个整数,u、v、w,每两个整数之间用一个空格隔开,表示从城市u到城市v有一条双向道路，设卡代价为w。
    数据保证输入的是一棵树,且根节点编号为1。

输出

其中共一行,包含一个整数,表示控制疫情所需要的最少设卡总代价。

输入示例

9
2 1 888
1 3 88
2 4 6
5 2 8
6 3 100
3 7 10
8 3 50
7 9 1

输出示例

102

其他说明

对于 60%的数据,2<n<10，对于 80%的数据,2<n<100，对于 100%的数据,2<n<100000

题解：直接写个最小割跑去吧。。。

Wa了两发是因为bfs得时候忘设vis[S]=true了。。。QAQ。。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
#define PAU putchar(' ')
#define ENT putchar('
')
using namespace std;
const int maxn=100000+10,maxm=200000+10,inf=-1u>>1;
int out[maxn];
struct isap{
    struct ted{int x,y,w;ted*nxt,*re;}adj[maxm],*fch[maxn],*ms,*cur[maxn],*ret[maxn];
    int d[maxn],gap[maxn],n,S,T;
    void init(int n){this->n=n;memset(d,-1,sizeof(d));ms=adj;return;}
    void add(int x,int y,int w){
        *ms=(ted){x,y,w,fch[x],ms+1};fch[x]=ms++;
        *ms=(ted){y,x,0,fch[y],ms-1};fch[y]=ms++;
        return;
    }
    void bfs(){
        queue<int>Q;Q.push(T);d[T]=0;
        while(!Q.empty()){
            int u=Q.front();Q.pop();
            for(ted*e=fch[u];e;e=e->nxt){
                int v=e->y;if(d[v]<0)d[v]=d[u]+1,Q.push(v);
            }
        } return;
    }
    int mxflow(int S,int T){
        this->S=S;this->T=T;bfs();ted*e;int k=S,flow=0;
        for(int i=1;i<=n;i++)gap[d[i]]++,cur[i]=fch[i];
        while(d[S]<n){
            if(k==T){
                int mi=inf,pos;for(int i=S;i!=T;i=cur[i]->y)if(cur[i]->w<mi)mi=cur[i]->w,pos=i;
                for(int i=S;i!=T;i=cur[i]->y)cur[i]->w-=mi,cur[i]->re->w+=mi;flow+=mi;k=pos;
            } for(e=cur[k];e;e=e->nxt)if(e->w&&d[k]==d[e->y]+1)break;
            if(e)cur[k]=e,ret[e->y]=e->re,k=e->y;
            else{if(--gap[d[k]]==0)break;cur[k]=fch[k];int mi=n;
                for(e=fch[k];e;e=e->nxt)if(e->w&&d[e->y]<mi)mi=d[e->y];
                d[k]=mi+1;gap[d[k]]++;if(k!=S)k=ret[k]->y;
            }
        } return flow;
    }
}sol;
struct ted{int x,y,w;ted*nxt;}adj[maxm],*fch[maxn],*ms=adj;
void add(int x,int y,int w){
    *ms=(ted){x,y,w,fch[x]};fch[x]=ms++;
    *ms=(ted){y,x,w,fch[y]};fch[y]=ms++;
    return;
}
bool vis[maxn];
void bfs(int S){
    queue<int>Q;Q.push(S);vis[S]=true;
    while(!Q.empty()){
        int u=Q.front();Q.pop();
        for(ted*e=fch[u];e;e=e->nxt){
            int v=e->y;if(!vis[v]){
                vis[v]=true;out[u]++;sol.add(u,v,e->w);Q.push(v);//printf("add:%d->%d %d
",u,v,e->w);system("pause");
            }
        }
    } return;
}
int n,m;queue<int>tset;
void makeG(){
    for(int i=1;i<=n;i++)if(!out[i])tset.push(i);//printf("tset:");
    while(!tset.empty()){
        int x=tset.front();tset.pop();sol.add(x,n+1,1e9);//printf("%d ",x);
    } return;
}
inline int read(){
    int x=0,sig=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') sig=-1;ch=getchar();}
    while(isdigit(ch)) x=10*x+ch-'0',ch=getchar();
    return x*=sig;
}
inline void write(int x){
    if(x==0){putchar('0');return;}if(x<0) putchar('-'),x=-x;
    int len=0,buf[15];while(x) buf[len++]=x%10,x/=10;
    for(int i=len-1;i>=0;i--) putchar(buf[i]+'0');return;
}
void init(){
    n=read();m=~-n;sol.init(n+1);int x,y,w;
    for(int i=1;i<=m;i++){
        x=read();y=read();w=read();add(x,y,w);
    }bfs(1);makeG();write(sol.mxflow(1,n+1));
    return;
}
void work(){
    return;
}
void print(){
    return;
}
int main(){
    init();work();print();return 0;
}