ACM-ICPC 2018 南京赛区网络预赛 L. Magical Girl Haze（分层图最短路）

Magical Girl Haze

题意:　n个点m条边 至多K条边的边权变为0 从1到n的最短路

AC code

/*
Magical Girl Haze
ACM-ICPC 2018 南京赛区网络预赛 L.
https://nanti.jisuanke.com/t/A1958
题意：n个点m条边 至多K条边的边权变为0 从1到n的最短路
解法：
分层图最短路
DP+spfa+优先队列
*/
#include <bits/stdc++.h>
using namespace std;
#define maxn 1000000
#define ll long long 
struct node{
	int u,v,w;
	int next;
}edge[maxn];
struct node1{
	int n,m;
	friend bool operator<(node1 a,node1 b)
	{
		return a.n>b.n;//按x从小到大排
	}
};
int head[maxn];
int dis[maxn][12];
int cnt,n,m,k;
void init(){
	memset(head,-1,sizeof head);
	memset(dis,0x3f3f3f3f,sizeof dis);
	cnt=0;
}
void add(int u,int v,int w){
	edge[cnt].u=u;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void spfa(){
	priority_queue<node1> q;
	q.push(node1{1,0});
	dis[1][0]=0;
	while(!q.empty()){
		node1 st=q.top();
		q.pop();
		for(int i=head[st.n];~i;i=edge[i].next){
			int v=edge[i].v;
			if(dis[v][st.m]>dis[st.n][st.m]+edge[i].w){
				dis[v][st.m]=dis[st.n][st.m]+edge[i].w;
				q.push(node1{v,st.m});
			}
			if(st.m+1<=k){
				if(dis[v][st.m+1]>dis[st.n][st.m]){
					dis[v][st.m+1]=dis[st.n][st.m];
					q.push(node1{v,st.m+1});
				}
			}
		}
	}
}
int main(){
	int T;
	scanf("%d", &T);
	while(T--){
		init();
		int u,v,w;
		scanf("%d %d %d", &n, &m, &k);
		for(int i=0;i<m;i++){
			scanf("%d %d %d", &u, &v, &w);
			add(u,v,w);
		}
		spfa();
		printf("%d\n", dis[n][k]);
	}
	return 0;
}

第一次做TEL code

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000000
struct node{
	int u,v,w;
	int next;
}edge[maxn];
int head[maxn];
bool vis[maxn];
int cnt,n,m,k;
long long ans;
multiset<int,greater<int>> mul;
void init(){
	memset(head,-1,sizeof head);
	memset(vis,false,sizeof vis);
	cnt=0;
	ans=0x3f3f3f3f;
	mul.clear();
}
void add(int u,int v,int w){
	edge[cnt].u=u;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int x,long long cnt){
	if(x==n){
		int i=0;
		for(set<int>::iterator it=mul.begin();it!=mul.end();it++){
			i++;
	    cnt-=(*it);
			if(i==k) break;
	  }
		ans=min(cnt,ans);
	}
	for(int i=head[x];~i;i=edge[i].next){
		if(!vis[edge[i].v]){
			cnt+=edge[i].w;
			mul.insert(edge[i].w);
			vis[edge[i].v]=true;
			dfs(edge[i].v,cnt);
			mul.erase(edge[i].w);
			vis[edge[i].v]=false;
			cnt-=edge[i].w;
		}
	}
}
int main(){
	int T;
	scanf("%d", &T);
	while(T--){
		init();
		int u,v,w;
		scanf("%d %d %d", &n, &m, &k);
		for(int i=0;i<m;i++){
			scanf("%d %d %d", &u, &v, &w);
			add(u,v,w);
		}
		dfs(1,0);
		printf("%lld\n", ans);
	}
	return 0;
}