CF1472G Moving to the Capital（bfs+dp）

题目链接

题意

在一张有向连通图中，边权均为1，每一个点到1号点的最短路径设为di，对于每个点有两种操作，

1. 沿着某条边，从第 i 个结点到第 jj个结点， di<dj；
2. 沿着某条边，从第 ii个结点到第 jj个结点，di​≥dj​；（每个点只能走一次）

对于每个点，问通过上述两操作能距1号点的最短距离。

思路

       di数组相当于以1为起点跑一次最短路。

       下面我们按di从大到小枚举维护答案，对于当前的点，只有通过2操作才能减小该点的di，且用一次2操作之后，便不能再用2操作，即不能再减小di，所以点j，且dj<=di,则ans[i]=min(ans[i],dj)。

       若选择点j，且dj>di,则ans[i]=min(ans[i],ans[j])，而由于我们是按di从大到小枚举的点，ans[j]我们已经维护了，直接用即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

struct node{
	int to,nex;
}edge[400010];
int head[200010],cnt;
int f[200010],dis[200010];
struct node2{
	int x,id;
}b[200010];
bool vis[200010];

void addedge(int x,int y){
	edge[++cnt].to=y;
	edge[cnt].nex=head[x];
	head[x]=cnt;
}

void bfs(){
	queue<int> q;
	q.push(1);
	dis[1]=0;
	vis[1]=true;
	while(!q.empty()){
		int p=q.front(); q.pop();
		for(int i=head[p];i!=-1;i=edge[i].nex){
			int y=edge[i].to;
			dis[y]=min(dis[y],dis[p]+1);
			if(!vis[y]){
				vis[y]=true; q.push(y);
			}
		}
	}
}

int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int _;
	cin>>_;
	while(_--){
		memset(head,-1,sizeof(head));
		memset(vis,false,sizeof(vis));
		memset(dis,0x3f,sizeof(dis));
		cnt=0;
		int n,m,x,y;
		cin>>n>>m;
		for(int i=1;i<=m;i++){
			cin>>x>>y;
			addedge(x,y);
		}
		bfs();  //bfs求di数组
		for(int i=1;i<=n;i++){
			f[i]=dis[i];
			b[i].id=i; b[i].x=dis[i];
		}
		sort(b+1,b+n+1,[](node2 a,node2 b){
			return a.x>b.x;
		});
		for(int i=1;i<=n;i++){
			int p=b[i].id;
			for(int j=head[p];j!=-1;j=edge[j].nex){
				int y=edge[j].to;
				if(dis[y]>dis[p]) f[p]=min(f[p],f[y]); //操作1
				else f[p]=min(f[p],dis[y]);  //操作2
			}
		}
		for(int i=1;i<=n;i++) cout<<f[i]<<" ";
		cout<<"\n";
	}
	return 0;
}

 要加油了...