2324: [ZJOI2011]营救皮卡丘

2324: [ZJOI2011]营救皮卡丘

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 2577  Solved: 1068
[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，你可以认为最终在小智的号召下，小智，小霞，小刚，小建，小遥，小胜，小光，艾莉丝，天桐，还有去日本旅游的黑猫警长，一同前去大战火箭队。

好题解。。。

实在想不明白，为什么循环队列会致T 泪

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=550,inf=0X3f3f3f3f;

int n,m,K,ans;

int dis[N][N];

void floyd()
{
	register int i,j,k;
	for(k=0;k<=n;++k)for(i=0;i<=n;++i)for(j=0;j<=n;++j)
	if(k<=i||k<=j)dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
}

int ecnt=1,last[N];
struct EDGE{int to,nt,val,ct;}e[401000];
inline void readd(int u,int v,int val,int ct)
{e[++ecnt]=(EDGE){v,last[u],val,ct};last[u]=ecnt;}
inline void add(int u,int v,int val,int ct)
{readd(u,v,val,ct);readd(v,u,0,-ct);}

int S=N-2,T=N-1;

void build()
{
	register int i,j;
	add(S,0,K,0);
	for(i=1;i<=n;++i)add(S,i,1,0);
	for(i=1;i<=n;++i)add(i+n,T,1,0);
	for(i=0;i<=n;++i)for(j=1+i;j<=n;++j)if(dis[i][j]<inf)
	add(i,j+n,1,dis[i][j]);
}

bool vis[N],inq[N];

int d[N],q[400100];

bool spfa()
{
	memset(d,0X3f,sizeof(d));
	memset(inq,0,sizeof(inq));
	memset(vis,0,sizeof(vis));
	register int u,i,head=0,tail=1;
	q[head]=S;inq[S]=1;d[S]=0;
	while(head<tail)
	{
		u=q[head++];inq[u]=0;
		for(i=last[u];i;i=e[i].nt)if(d[e[i].to]>d[u]+e[i].ct&&e[i].val)
		{
			d[e[i].to]=e[i].ct+d[u];
			if(!inq[e[i].to])inq[e[i].to]=1,q[tail++]=e[i].to;
		}
	}
	return d[T]<inf;
}

int dfs(int u,int lim)
{
	if(T==u){ans+=d[T]*lim;return lim;}
	vis[u]=1;
	int tmp=0,res=0;
	for(int i=last[u];i;i=e[i].nt)
	if(d[e[i].to]==d[u]+e[i].ct&&e[i].val&&!vis[e[i].to])
	{
		tmp=dfs(e[i].to,min(e[i].val,lim));
		lim-=tmp;res+=tmp;e[i].val-=tmp;e[i^1].val+=tmp;
		if(!tmp)d[e[i].to]=-1;if(!lim)break;
	}
	return res;
}

void mincf()
{while(spfa())dfs(S,inf);}

int main()
{
	n=read();m=read();K=read();
	register int i,u,v,val;
	memset(dis,0X3f,sizeof(dis));
	for(i=0;i<=n;++i)dis[i][i]=0;
	for(i=1;i<=m;++i)
	{
		u=read();v=read();val=read();
		dis[u][v]=dis[v][u]=min(dis[u][v],val);
	}
	floyd();build();mincf();
	print(ans);puts("");
	return 0;
}
/*
3 4 2
0 1 1
1 2 1
2 3 100
0 3 1

3
*/