题意:题意比较容易,有K条边是免费,求1到N的最短路。
思路:选择一条M段路从1走到N,其中M<=K是最优的解。所以这里是要统计1到N有多少段路与K比较即可,当M>K时,M要尽可能靠近K,并且边的权值要小。
通过枚举边大小E,大于E的边赋值1,小于等于1的边赋值0。二分边权值,跑最短路。具体看代码。
AC代码:
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=1010;
const int INF=999999999;
int map[maxn][maxn];
struct edge
{
int u,v,l;
};
struct edge E[maxn*10];
bool cmp(edge a,edge b)
{
return a.l<b.l;
}
void build(int n,int p,int l)
{
int i,j;
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
map[i][j]=INF;
}
}
for(i=0;i<p;i++)
{
int a,b,c;
a=E[i].u;
b=E[i].v;
c=E[i].l;
if(c>l)//大于L赋值为1
map[a][b]=map[b][a]=1;
else
map[a][b]=map[b][a]=0;
}
}
int Dijkstra(int s,int n)
{
int i,j,k,t;
bool vis[maxn];
int dis[maxn];
for(i=1;i<=n;i++)
{
dis[i]=(s==i?0:map[s][i]);
vis[i]=false;
}
vis[s]=true;
dis[s]=0;
for(i=1;i<n;i++)
{
t=INF;
for(j=1;j<=n;j++)
{
if(!vis[j] && dis[j]<t)
{
t=dis[j];
k=j;
}
}
if(t==INF)
break;
vis[k]=true;
for(j=1;j<=n;j++)
{
if(!vis[j] && dis[j]>dis[k]+map[k][j])
dis[j]=dis[k]+map[k][j];
}
}
return dis[n];
}
int bfind(int p,int n,int k)
{
int l,r,m;
l=0;
r=p-1;
while(l<=r)
{
m=(l+r)/2;
build(n,p,E[m].l);
if(Dijkstra(1,n)<=k)//DIJ<=K说明边数少了,缩小上界。
r=m-1;
else
l=m+1;
}
return l;//最小的边
}
int main()
{
int i;
int n,p,k;
while(scanf("%d %d %d",&n,&p,&k)!=EOF)
{
for(i=0;i<p;i++)
scanf("%d %d %d",&E[i].u,&E[i].v,&E[i].l);
sort(E,E+p,cmp);
build(n,p,0);
int ans;
ans=Dijkstra(1,n);
if(ans==INF) //判断是否能到达。
printf("%d\n",-1);
else
{
if(ans<=k)
printf("%d\n",0);
else
{
int indx=bfind(p,n,k);
printf("%d\n",E[indx].l);
}
}
}
return 0;
}