大逃离

大逃离

基本就是模板题

加了1个限制，能到达点的个数不小于k

注意：出度+入度≠能到达点的个数，此时n的范围又很小，可以开个二维数组标记能到达的点，且同时统计数量

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>

using namespace std;
int n,m,k;
const long long maxn=5e3+10,maxm=maxn*21,inf=0x3f3f3f3f3f3f;
int head[maxm<<1];
int cnt=0;
int dd[maxn][maxn];
int d[maxn];
long long dis1[maxn],dis2[maxn];

struct node{
	int v,next,w;
}e[maxm<<1];
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dj(){
	for(int i=1;i<=n;++i) dis1[i]=dis2[i]=inf;
	priority_queue<pair<long long ,long long> >q;
	dis1[1]=0;
	q.push(make_pair(0,1)); 
	while(!q.empty()){
		long long  u=q.top().second;
		long long dis=-q.top().first;
		q.pop();
		for(int i=head[u];i;i=e[i].next){
			int v=e[i].v,w=e[i].w;
			if(d[v]<k && v!=1 && v!=n) continue;
			if(dis1[v]>dis+w){
				dis2[v]=dis1[v];
				dis1[v]=dis+w;
				q.push(make_pair(-dis1[v],v));
			} 
			if(dis2[v]>dis+w && dis+w>dis1[v]){
				dis2[v]=dis+w;
				q.push(make_pair(-dis2[v],v)); 
			}
		}
	}
} 
int main(){
	cin>>n>>m>>k;
	for(int i=1;i<=m;++i){
		int u,v,w;
		cin>>u>>v>>w;
		if(u!=v){
			if(dd[u][v]!=1) ++d[v],dd[u][v]=1;
			if(dd[v][u]!=1) ++d[u],dd[v][u]=1;
			
			add(u,v,w),add(v,u,w);
		}
	}
	dj();
	if(dis2[n]==inf) puts("-1");
	else cout<<dis2[n]<<endl;
	return 0; 
}

ZFY AK IOI