题意:
给出一个有向图, 1 1 1为起点, d i d_i di 为起点到其他各个点的最短路,现在给出两种前进方式:
1. 1. 1.如果 i i i 与 j j j 相连, d i < d j d_i<d_j di<dj ,那么点 i i i 可以前往点 j j j 。
2. 2. 2. 如果 i i i 与 j j j 相连, d i ≥ d j d_i \geq d_j di≥dj ,那么点 i i i 可以前往点 j j j ,但这个方式只能进行一次。
求从每个点出发,按照上述两种方式前进,能到达点的最小 d i d_i di 是多少。
题解:
一开始必定是先用 d j dj dj 求出每个点的最短路。
然后一开始我采取的做法是从起点开始 d f s dfs dfs ,设 d p [ i ] dp[i] dp[i]数组表示从点 i i i 出发的最小的 d d d,然后用直接相连的点去更新当前点的 d p dp dp值,但是这样在环比较多的时候很容易出错,达不到最优解,最后也光荣的 w r o n g a n s w e r 2 wrong\quad answer \quad 2 wronganswer2 。
那么正解要咋写呢?从题目条件出发去想,第二种方式是走 d d d 小的,然后只能用一次,说明假如选了这个点,那么后面不管怎么走,最小的一定是这个 d d d,那么 d p [ i ] = m i n ( d p [ i ] , d j ) dp[i]=min(dp[i],d_j) dp[i]=min(dp[i],dj) ,那么假如要走大的 d d d,那么 d p [ i ] = m i n ( d p [ i ] , d p [ j ] ) dp[i]=min(dp[i],dp[j]) dp[i]=min(dp[i],dp[j]) ,
怎么求 d p [ j ] dp[j] dp[j] ?从大往小递推即可,即对点按照 d i d_i di 排序从大到小排序,先求大的,那么小的自然能递推出来。
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#include<ctime>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int mod=1e9+7;
const int MAXN=2e5+5;
const int inf=0x3f3f3f3f;
int head[MAXN];
int vis[MAXN];
int dis[MAXN];
int u[MAXN],v[MAXN];
int cnt;
int dp[MAXN];
struct edge
{
int to;
int cost;
int next;
}e[MAXN*2];
struct node
{
int pos;
ll val;
friend bool operator<(node a,node b)
{
return a.val>b.val;
}
};
void add(int u,int v,int w)
{
e[cnt].to=v;
e[cnt].cost=w;
e[cnt].next=head[u];
head[u]=cnt++;
}
void solve(int s)
{
memset(vis,0,sizeof vis);
memset(dis,0x3f3f3f,sizeof dis);
priority_queue<node> q;
q.push(node{s,0});
dis[s]=0;
while(!q.empty())
{
node now=q.top();
q.pop();
if(vis[now.pos]) continue;
vis[now.pos]=1;
for(int i=head[now.pos];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(dis[v]>now.val+e[i].cost)
{
dis[v]=now.val+e[i].cost;
q.push(node{v,dis[v]});
}
}
}
}
struct N
{
int id;
int d;
}f[MAXN];
bool cmp(N x,N y)
{
return x.d>y.d;
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n,m;
cin>>n>>m;
cnt=0;
for(int i=1;i<=n;i++)
{
head[i]=-1;
}
for(int i=1;i<=m;i++)
{
cin>>u[i]>>v[i];
add(u[i],v[i],1);
}
solve(1);
for(int i=1;i<=n;i++)
{
f[i].id=i;
f[i].d=dis[i];
dp[i]=dis[i];
}
sort(f+1,f+1+n,cmp);
for(int i=1;i<=n;i++)
{
int id=f[i].id;
for(int j=head[id];j!=-1;j=e[j].next)
{
int vv=e[j].to;
if(dis[vv]>dis[id]) dp[id]=min(dp[id],dp[vv]);
else dp[id]=min(dp[id],dis[vv]);
}
}
for(int i=1;i<=n;i++)
{
cout<<dp[i]<<" ";
}
cout<<endl;
}
}