Description
每天,农夫John需要经过一些道路去检查牛棚N里面的牛. 农场上有M(1<=M<=50,000)条双向泥土道路,编号为1..M. 道路i连接牛棚P1_i和P2_i (1 <= P1_i <= N; 1 <= P2_i<= N). John需要T_i (1 <= T_i <= 1,000,000)时间单位用道路i从P1_i走到P2_i或者从P2_i 走到P1_i 他想更新一些路经来减少每天花在路上的时间.具体地说,他想更新K (1 <= K <= 20)条路经,将它们所须时间减为0.帮助FJ选择哪些路经需要更新使得从1到N的时间尽量少.
Input
* 第一行: 三个空格分开的数: N, M, 和 K * 第2..M+1行: 第i+1行有三个空格分开的数:P1_i, P2_i, 和 T_i
Output
* 第一行: 更新最多K条路经后的最短路经长度.
Sample Input
4 4 1
1 2 10
2 4 10
1 3 1
3 4 100
1 2 10
2 4 10
1 3 1
3 4 100
Sample Output
1
HINT
K是1; 更新道路3->4使得从3到4的时间由100减少到0. 最新最短路经是1->3->4,总用时为1单位. N<=10000
题意是n个点m条边,可以把最多k条边的权值改成0,求1到n的最短路
很容易想到用分层的办法做
构造0~k层共k+1层的图,每一层都有n个点。第i层表示还能用i次修改权值的情况下,S到每个点的距离最小值
然后连边的时候跨层的连边权为0,同层的连边权为v
然后最短路
可以当做dijkstra+堆的模板来用了
#include<cstdio> #include<iostream> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<deque> #include<set> #include<map> #include<ctime> #define inf 2147483647 #define S (1+k*n) #define T n #define N 500010 #define pa pair<int,int> #define mkp(x,y) make_pair(x,y) using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } struct edge{int to,next,v;}e[5000010]; priority_queue<pa,vector<pa>,greater<pa> >q; int head[N]; bool vis[N]; int dist[N]; int n,m,k,sx,sy,cnt; inline void ins(int u,int v,int w) { e[++cnt].to=v; e[cnt].v=w; e[cnt].next=head[u]; head[u]=cnt; } inline void insert(int u,int v,int w) { ins(u,v,w); ins(v,u,w); } inline void dijkstra() { for (int i=1;i<=n*k+n;i++)dist[i]=inf; dist[S]=0;q.push(mkp(0,S)); while (!q.empty()) { int now=q.top().second;q.pop(); if(vis[now])continue;vis[now]=1; for(int i=head[now];i;i=e[i].next) if(dist[now]+e[i].v<dist[e[i].to]) { dist[e[i].to]=dist[now]+e[i].v; q.push(mkp(dist[e[i].to],e[i].to)); } } } int main() { n=read();m=read();k=read(); for (int i=n+1;i<=n*k+n;i++)ins(i,i-n,0); for (int i=1;i<=m;i++) { int x=read(),y=read(),z=read(); for(int j=1;j<=k;j++) { sx=x+j*n; sy=y+j*n; insert(sx,sy,z); ins(sx,sy-n,0); ins(sy,sx-n,0); } insert(x,y,z); } dijkstra(); if (dist[T]!=inf)printf("%d ",dist[T]); else printf("NO SOLUTION"); }