牛客练习赛 50 D tokitsukaze and Event （思维 + 双向最短路）

大致题意

给定 n 个点 m 条边的无向图，每条边有两个边权 ai, bi。给定起点和终点s,t。一开始只能选择走边 ai ，你可以选择在某个点转换状态，在这之后你只能走边 bi ，且只能转化状态一次。要求从 s 走到 t 的最短路。当然没这么简单，现在规定这个游戏的难度 p ，当难度为 p 时,你可以选择所有 点的标号 i >=p 的点进行状态转化，否则不能转换状态。现在要求在难度 1到 n 的情况下对应的 s 到 t 的最短路。
n<=1e5 ， ai <=1e9 , bi<=1e9

思路

容易想到整条路径分为两段，假设转换点为 x ，那么路径分为从 s 到 x 走边 ai，还有从 x 到 t 走边 bi。所以考虑求出 s 走 ai 的单源点最短路，还有从 t 走bi 的单源点最短路，然后 O(n) 枚举每一个点 i 作为 转换点 的最短路。这正相当于当难度为 i 时 用 i 点作为转换点的最短路。因为当难度为 i 时，所有 j >=i 也符合当前难度。所有再做一个后缀最小值。

代码

我记得djistra也是要加标记的，以前一直不加，这里尝试加了一下，不知道对不对，但是加完运行时间变长了…

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define mod 998244353
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int n,m,s,t;
struct node{int v,w[2],nxt;}ed[maxn<<1];
int head[maxn],tot;
void add(int x,int y,int z1,int z2){ed[++tot]={y,{z1,z2},head[x]};head[x]=tot;}
ll dis[2][maxn];
bool vis[maxn];
priority_queue<pair<ll,int> >q;
void djistra(int st,int id){
    while(!q.empty())q.pop();
    inc(i,1,n){dis[id][i]=inf;vis[i]=0;}
    dis[id][st]=0;
    q.push(make_pair(0,st));
    while(!q.empty()){
        int u=q.top().second;q.pop();
        if(vis[u])continue;
        vis[u]=1;
        for(int i=head[u];i;i=ed[i].nxt){
            int v=ed[i].v;
            if(dis[id][v]>dis[id][u]+ed[i].w[id]){
                dis[id][v]=dis[id][u]+ed[i].w[id];
                q.push(make_pair(-dis[id][v],v));
            }
        }
    }
}
ll ans[maxn];
int main()
{
    n=read();m=read();
    int x,y,a,b;
    inc(i,1,m){
        x=read();y=read();a=read();b=read();
        add(x,y,a,b);add(y,x,a,b);
    }
    s=read();t=read();
    djistra(s,0);djistra(t,1);
    inc(i,1,n)ans[i]=inf;
    inc(i,1,n)ans[i]=min(ans[i],dis[0][i]+dis[1][i]);
    dec(i,n-1,1)ans[i]=min(ans[i],ans[i+1]);
    inc(i,1,n)printf("%lld\n",ans[i]);
    return 0;
}