2324: [ZJOI2011]营救皮卡丘

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

于是只好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;
}