【jzoj4597】【现世斩】【平衡树】【set】

题目大意

给出有权无向连通图，有一次机会使与一个点相连的边的权变为一，求最短路。

题解

显然的想法是学习gdoi2016的一题，分层单源最短路，设f[i][0,2],表示没有用机会，在前驱用机会，没有机会，dijset优化就可以了。

还有一种方法，只有一次机会，影响很小，起点终点都做一次最短路，枚举在哪里用，扫一下边，每条边只被扫2次，复杂度有保证。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=100000,maxm=500000;
int n,m,gra,to[maxm*2+10],val[maxm*2+10],next[maxm*2+10],begin[maxn+10];
void insert(int u,int v,int w){
    to[++gra]=v;
    val[gra]=w;
    next[gra]=begin[u];
    begin[u]=gra;
}
struct rec{
    int pon,sta;long long dis;
    friend bool operator<(rec i,rec j){
        return (i.dis<j.dis)||((i.dis==j.dis)&&(i.pon<j.pon))||((i.dis==j.dis)&&(i.pon==j.pon)&&(i.sta<j.sta));
    }
};
multiset<rec>s;
long long f[maxn+10][10];
void dij(){
    memset(f,63,sizeof(f));
    s.insert({1,0,0});
    s.insert({1,1,0});
    rec now=*s.begin();
    for(;!s.empty();){
        for(rec now=*s.begin();(!s.empty())&&(f[now.pon][now.sta]<=now.dis);s.erase(s.begin()),now=*s.begin());
        if(s.empty())break;
        rec now=*s.begin();f[now.pon][now.sta]=now.dis;s.erase(s.begin());
        for(int i=begin[now.pon];i;i=next[i])
            if(now.sta==1){
                if(f[to[i]][2]>now.dis+1)s.insert({to[i],2,now.dis+1});
            }else if(now.sta==0){
                if(f[to[i]][0]>now.dis+val[i])s.insert({to[i],0,now.dis+val[i]});
                if(f[to[i]][1]>now.dis+1)s.insert({to[i],1,now.dis+1});
            }else{
                if(f[to[i]][2]>now.dis+val[i])s.insert({to[i],2,now.dis+val[i]});
            }
    }
}
int main(){
    freopen("cut.in","r",stdin);
    freopen("cut.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,m){
        int u,v,w;scanf("%d%d%d",&u,&v,&w);
        insert(u,v,w);insert(v,u,w);
    }
    dij();
    printf("%lld",min(f[n][0],min(f[n][1],f[n][2])));
    return 0;
}