2324: [ZJOI2011]营救皮卡丘

来源：互联网发布：mac终端编辑保存退出编辑：程序博客网时间：2024/04/30 04:07

一开始想的是边建图边跑费用流，不过这样贪心好像不行。

于是只好floyd预处理后再建图了。

floyd的时候需要过渡节点k小于i且k小于j，然后由于题目的关系可以砍掉一些边（快一点）。

本来以为这题卡普通费用流，因为TLE了两次，结果发现是因为数组开小了QAQ

 #include<iostream>#include<cstdio>#include<cstring>#include<queue>using namespace std;const int inf=1e9;struct Edge{int from,to,next,v,c;}e[100005];int g[155][155];int head[310],from[310],d[310],cnt=1,n;bool inq[310];void ins(int u,int v,int w,int c){    cnt++;e[cnt].from=u;e[cnt].to=v;    e[cnt].next=head[u];head[u]=cnt;    e[cnt].v=w;e[cnt].c=c;}void insert(int u,int v,int w,int c){    ins(u,v,w,c);ins(v,u,0,-c);}bool spfa(int s,int t,int &cost){    memset(d,0x3f,sizeof(d));    d[s]=0;queue<int>q;q.push(s);    while(!q.empty()){        int u=q.front();q.pop();inq[u]=false;        for(int i=head[u];i;i=e[i].next)        if(e[i].v&&d[e[i].to]>d[u]+e[i].c){            d[e[i].to]=d[u]+e[i].c;            from[e[i].to]=i;            if(!inq[e[i].to]){inq[e[i].to]=true;q.push(e[i].to);}        }    }    if(d[t]>=inf)return false;    int x=inf;    for(int i=from[t];i;i=from[e[i].from])x=min(x,e[i].v);    for(int i=from[t];i;i=from[e[i].from]){        e[i].v-=x;e[i^1].v+=x;cost+=x*e[i].c;    }    return true;}int mcmf(int s,int t){int cost=0;while(spfa(s,t,cost));return cost;}int main(){    int m,k;scanf("%d%d%d",&n,&m,&k);    int S=n+n+1,T=S+1;    insert(S,0,k,0);    int u,v,w;    memset(g,0x3f,sizeof(g));    for(int i=1;i<=m;i++){        scanf("%d%d%d",&u,&v,&w);        g[u][v]=min(g[u][v],w);        g[v][u]=min(g[v][u],w);    }    for(int h=0;h<=n;h++)    for(int i=0;i<=n;i++)    for(int j=0;j<=n;j++)    if(i!=j){        if(g[i][h]<inf&&g[h][j]<inf)        g[i][j]=min(g[i][j],g[i][h]+g[h][j]);        if(h==j&&i<j&&g[i][j]<inf)        insert(i==0?0:i+n,j,inf,g[i][j]);    }    for(int i=1;i<=n;i++){        insert(S,i+n,1,0);        insert(i,T,1,0);    }    printf("%d",mcmf(S,T));    return 0;}

0 0