题目大意:
n个点,m条边(n<=10000 m<=50000)可以任意将k(k<=10)条边的值改为0,求S到T的最短路。
题目分析:
把一个图拆成k+1个图,连边的时候相同层之间连正常权值,然后从当前层向下一层连一条权值为0的单向边。
代表有k次机会走一条为0的边到下一层。
直接跑Dijkstra即可。
什么?你说spfa??
……
代码如下:
#include <cstdio>
#include <cstring>
#include <queue>
#define N 120000
#define M 2200000
using namespace std;
typedef long long LL;
int n,m,K,S,T;
int fir[N],nes[M],v[M],q[M],tot=1;
LL f[N];
bool vis[N];
struct point{
LL val,;
int pos;
point(LL val,int pos):val(val),pos(pos){}
bool operator < (const point &c) const { return val>c.val; }
};
void edge(int x,int y,int z)
{
v[++tot]=y;
q[tot]=z;
nes[tot]=fir[x];
fir[x]=tot;
}
#define Edge(x,y,z) edge(x,y,z),edge(y,x,z)
void Dijkstra()
{
static int c;
memset(f,0x3f,sizeof(f));
priority_queue<point> dl;
vis[S]=true;
for(int t=fir[S];t;t=nes[t])
if(v[t]!=S) dl.push(point(q[t],v[t]));
int cnt=1,sum=K*n+n;
for(;!dl.empty();dl.pop())
{
if(vis[dl.top().pos]) continue;
c=dl.top().pos; f[c]=dl.top().val;
vis[c]=true;
for(int t=fir[c];t;t=nes[t])
if(!vis[v[t]]) dl.push(point(f[c]+q[t],v[t]));
if(++cnt==sum) break;
}
}
int main()
{
scanf("%d%d%d",&n,&m,&K);
scanf("%d%d",&S,&T);
for(int i=1,x,y,z;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&z);
Edge(x,y,z);
for(int i=1;i<=K;i++)
{
Edge(i*n+x,i*n+y,z);
edge((i-1)*n+x,i*n+y,0);
edge((i-1)*n+y,i*n+x,0);
}
}
Dijkstra();
LL ans=0x3f3f3f3f3f3f3f3fll;
for(int i=0;i<=K;i++)
ans=f[i*n+T]<ans?f[i*n+T]:ans;
printf("%lld\n",ans);
return 0;
}