bzoj 2324 [ZJOI2011]营救皮卡丘

2324: [ZJOI2011]营救皮卡丘

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2517 Solved: 1044
[Submit][Status][Discuss]
Description

皮卡丘被火箭队用邪恶的计谋抢走了！这三个坏家伙还给小智留下了赤果果的挑衅！为了皮卡丘，也为了正义，小智和他的朋友们义不容辞的踏上了营救皮卡丘的道路。
火箭队一共有N个据点，据点之间存在M条双向道路。据点分别从1到N标号。小智一行K人从真新镇出发，营救被困在N号据点的皮卡丘。为了方便起见，我们将真新镇视为0号据点，一开始K个人都在0号点。
由于火箭队的重重布防，要想摧毁K号据点，必须按照顺序先摧毁1到K-1号据点，并且，如果K-1号据点没有被摧毁，由于防御的连锁性，小智一行任何一个人进入据点K，都会被发现，并产生严重后果。因此，在K-1号据点被摧毁之前，任何人是不能够经过K号据点的。
为了简化问题，我们忽略战斗环节，小智一行任何一个人经过K号据点即认为K号据点被摧毁。被摧毁的据点依然是可以被经过的。
K个人是可以分头行动的，只要有任何一个人在K-1号据点被摧毁之后，经过K号据点，K号据点就被摧毁了。显然的，只要N号据点被摧毁，皮卡丘就得救了。
野外的道路是不安全的，因此小智一行希望在摧毁N号据点救出皮卡丘的同时，使得K个人所经过的道路的长度总和最少。
请你帮助小智设计一个最佳的营救方案吧！
Input

第一行包含三个正整数N，M，K。表示一共有N+1个据点，分别从0到N编号，以及M条无向边。一开始小智一行共K个人均位于0号点。
接下来M行，每行三个非负整数，第i行的整数为Ai，Bi，Li。表示存在一条从Ai号据点到Bi号据点的长度为Li的道路。
Output

仅包含一个整数S，为营救皮卡丘所需要经过的最小的道路总和。
Sample Input

3 4 2

0 1 1

1 2 1

2 3 100

0 3 1
Sample Output

3

【样例说明】

小智和小霞一起前去营救皮卡丘。在最优方案中，小智先从真新镇前往1号点，接着前往2号据点。当小智成功摧毁2号据点之后，小霞从真新镇出发直接前往3号据点，救出皮卡丘。
HINT

对于100%的数据满足N ≤ 150, M ≤ 20 000, 1 ≤ K ≤ 10, Li ≤ 10 000, 保证小智一行一定能够救出皮卡丘。至于为什么K ≤ 10，你可以认为最终在小智的号召下，小智，小霞，小刚，小建，小遥，小胜，小光，艾莉丝，天桐，还有去日本旅游的黑猫警长，一同前去大战火箭队。

Source

Day2

---

【分析】

有上下界的费用流…?
第一次接触这个有(qi)趣(pa)的东西。
大概套路是将下界边费用设为负值，这样保证一定会经过，求出答案以后再把这些负值加回来。
原理貌似比较简单…QAQ

---

【代码】

//bzoj 2324 [ZJOI2011]营救皮卡丘
#include<bits/stdc++.h>
#define ll long long
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=400005;
const int inf=1e9+7;
bool vis[mxn];
queue <int> q;
ll ans,dis[mxn];
int cnt,n,m,k,S,T;
int len[155][155],pre[mxn],head[mxn];
struct edge {int from,to,w,flow,next;} f[mxn<<1];
inline int read()
{
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x;
}
inline void add(int u,int v,int w,int flow)
{
    f[++cnt]=(edge){u,v,w,flow,head[u]},head[u]=cnt;
    f[++cnt]=(edge){v,u,-w,0,head[v]},head[v]=cnt;
}
inline bool spfa()
{
    memset(dis,0x3f,sizeof dis);
    memset(pre,0,sizeof pre);
    dis[S]=0,q.push(S);
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=0;
        for(int i=head[u];i;i=f[i].next)
        {
            int v=f[i].to;
            if(dis[v]>dis[u]+f[i].w && f[i].flow>0)
            {
                dis[v]=dis[u]+f[i].w,pre[v]=i;
                if(!vis[v]) vis[v]=1,q.push(v);
            }
        }
    }
    return pre[T];
}
inline void maxflow()
{
    while(spfa())
    {
        int tmp=inf;
        for(int i=pre[T];i;i=pre[f[i].from])
          tmp=min(tmp,f[i].flow);
        ans+=dis[T]*tmp;
        for(int i=pre[T];i;i=pre[f[i].from])
        {
            f[i].flow-=tmp;
            if(i&1) f[i+1].flow+=tmp;
            else f[i-1].flow+=tmp;
        }
    }
}
int main()
{
    int i,j,u,v,w,p;
    memset(len,0x3f,sizeof len);
    n=read(),m=read(),k=read();
    S=n+n+1,T=n+n+2;
    fo(i,1,n) len[i][i]=0;
    while(m--)
    {
        u=read(),v=read(),w=read();
        len[u][v]=len[v][u]=min(len[u][v],w);
    }
    fo(p,0,n) fo(i,0,n) fo(j,0,n) if(p<=i || p<=j)
      len[i][j]=min(len[i][j],len[i][p]+len[p][j]);
    fo(i,1,n) add(i+n,i,-inf,1),add(i+n,i,0,inf),add(i,T,0,1);
    fo(i,0,n) fo(j,i+1,n) add(i,j+n,len[i][j],1);add(S,0,0,k);
    maxflow();
    printf("%lld\n",ans+(ll)n*inf);
    return 0;
}