2324: [ZJOI2011]营救皮卡丘
Time Limit: 10 Sec Memory Limit: 256 MBSubmit: 2406 Solved: 985
[ 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
0 1 1
1 2 1
2 3 100
0 3 1
Sample Output
【样例说明】
小智和小霞一起前去营救皮卡丘。在最优方案中,小智先从真新镇前往1号点,接着前往2号据点。当小智成功摧毁2号据点之后,小霞从真新镇出发直接前往3号据点,救出皮卡丘。
HINT
对于100%的数据满足N ≤ 150, M ≤ 20 000, 1 ≤ K ≤ 10, Li ≤ 10 000, 保证小智一行一定能够救出皮卡丘。至于为什么K ≤ 10,你可以认为最终在小智的号召下,小智,小霞,小刚,小建,小遥,小胜,小光,艾莉丝,天桐,还有去日本旅游的黑猫警长,一同前去大战火箭队。
Source
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<bitset>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;
const int maxn = 505;
const int maxm = 1E5 + 10;
const int INF = 1E9 + 233;
struct E{
int to,cap,flow,cost; E(){}
E(int to,int cap,int flow,int cost):
to(to),cap(cap),flow(flow),cost(cost){}
}edgs[maxm];
int n,m,K,s,t,S,T,cnt,Flow[maxn],Cost[maxn],from[maxn],dis[maxn][maxn];
bool vis[maxn];
vector <int> v[maxn];
queue <int> Q;
int getint()
{
char ch = getchar(); int ret = 0;
while (ch < '0' || '9' < ch) ch = getchar();
while ('0' <= ch && ch <= '9')
ret = ret * 10 + ch - '0',ch = getchar();
return ret;
}
void Add(int x,int y,int cap,int cost)
{
v[x].push_back(cnt); edgs[cnt++] = E(y,cap,0,cost);
v[y].push_back(cnt); edgs[cnt++] = E(x,0,0,-cost);
}
bool SPFA()
{
for (int i = 0; i <= T; i++) Cost[i] = INF;
Q.push(S); Flow[S] = INF; Cost[S] = 0; vis[S] = 1;
while (!Q.empty())
{
int k = Q.front(); Q.pop(); vis[k] = 0;
for (int i = 0; i < v[k].size(); i++)
{
E e = edgs[v[k][i]];
if (e.cap == e.flow) continue;
if (Cost[e.to] > Cost[k] + e.cost)
{
from[e.to] = v[k][i];
Cost[e.to] = Cost[k] + e.cost;
Flow[e.to] = min(Flow[k],e.cap - e.flow);
if (!vis[e.to]) vis[e.to] = 1,Q.push(e.to);
}
}
}
return Cost[T] != INF;
}
void Floyed()
{
for (int i = 0; i <= n; i++)
for (int j = 0; j <= n; j++)
dis[i][j] = INF;
for (int i = 0; i <= n; i++) dis[i][i] = 0;
while (m--)
{
int x = getint(),y,w;
y = getint(); w = getint();
dis[x][y] = min(dis[x][y],w);
dis[y][x] = min(dis[y][x],w);
}
for (int k = 0; k <= n; k++)
for (int x = 0; x <= n; x++)
for (int y = 0; y <= n; y++)
if (k <= x || k <= y) dis[x][y] = min(dis[x][y],dis[x][k] + dis[k][y]);
}
int main()
{
#ifdef DMC
freopen("DMC.txt","r",stdin);
#endif
n = getint(); m = getint(); K = getint();
s = 2 * n + 1; t = s + 1; S = t + 1; T = S + 1;
Floyed(); Add(s,0,K,0); Add(t,s,INF,0);
for (int i = 1; i <= n; i++)
{
Add(i,T,1,0); Add(S,i + n,1,0);
Add(i + n,t,INF,0); Add(i,i + n,INF,0);
}
for (int i = 0; i < n; i++)
for (int j = i + 1; j <= n; j++)
if (dis[i][j] != INF) Add(i ? i + n : i,j,INF,dis[i][j]);
int Ans = 0;
while (SPFA())
{
Ans += Flow[T] * Cost[T];
for (int x = T; x != S; x = edgs[from[x]^1].to)
{
edgs[from[x]].flow += Flow[T];
edgs[from[x]^1].flow -= Flow[T];
}
}
cout << Ans << endl;
return 0;
}