题解 bzoj 4398福慧双修（二进制分组）

二进制分组，算个小技巧

bzoj 4398福慧双修

给一张图，同一条边不同方向权值不同，一条边只能走一次，求从1号点出发再回到1号点的最短路
一开始没注意一条边只能走一次这个限制，打了个从一号点相邻节点为原点的dij，样例就挂了
其实就是要从这个错误思路上改进
对于不与1号点相接的边，权值为正，肯定不会重复走，所以这个条件可以忽略
考虑1号点相邻的点，走出第一步后所在的点，和走回1号点前的那个点不能相同
设这两个点编号为\(i\),\(j\),则\(i\),\(j\)的二进制至少有一位不同
所以用二进制分组其实这个方法也是看来题解之后才知道的，自己想真很难想出来
枚举每个二进制位，虚拟一个源点和汇点，这位为0/1的点分成两组，分别与源/汇点相连，跑两次dij，取\(dis[1]\)的最小值
复杂度高于那种构造新图的方式，bzoj上测的总时间2s+，肯定是倒数，但想起来也能简单一些
code.

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<iomanip>
#include<cstring>
#define R register
#define EN std::puts("")
#define LL long long
inline int read(){
    int x=0,y=1;
    char c=std::getchar();
    while(c<'0'||c>'9'){if(c=='-') y=0;c=std::getchar();}
    while(c>='0'&&c<='9'){x=x*10+(c^48);c=std::getchar();}
    return y?x:-x;
}
int n,m;
int fir[40006],nex[200006],to[200006],w[200006],tot;
int dui[40006],size;
int dis[40006],in[40006];
inline void push(int x){
    dui[size++]=x;
    R int i=size-1,fa;
    while(i){
        fa=i>>1;
        if(dis[dui[fa]]<=dis[dui[i]]) return;
        std::swap(dui[fa],dui[i]);i=fa;
    }
}
inline int pop(){
    int ret=dui[0];dui[0]=dui[--size];
    R int i=0,ls,rs;
    while((i<<1)<size){
        ls=i<<1;rs=ls|1;
        if(rs<size&&dis[dui[rs]]<dis[dui[ls]]) ls=rs;
        if(dis[dui[ls]]>=dis[dui[i]]) break;
        std::swap(dui[ls],dui[i]);i=ls;
    }
    return ret;
}
inline int dij(int bit,int panduan){
    std::memset(dis,0x3f,sizeof dis);
    for(R int i=fir[1];i;i=nex[i])if((to[i]&bit)==panduan){
        push(to[i]);dis[to[i]]=w[i];in[to[i]]=1;
    }
    while(size){
        R int u=pop();in[u]=0;
        for(R int i=fir[u];i;i=nex[i]){
            R int v=to[i];
            if(v==1&&(u&bit)==panduan) continue;
            if(dis[v]>dis[u]+w[i]){
                dis[v]=dis[u]+w[i];
                if(!in[v]) push(v),in[v]=1;
            }
        }
    }
    return dis[1];
}
inline void add(int x,int y,int z){
    to[++tot]=y;w[tot]=z;
    nex[tot]=fir[x];fir[x]=tot;
}
int work(){
    R int ret=0x3f3f3f3f;
    R int tmp=0;int nn=n;
    while(nn) tmp++,nn>>=1;
    for(R int i=0;i<tmp;i++)
        {ret=std::min(ret,dij(1<<i,1<<i)),ret=std::min(ret,dij(1<<i,0));}
    return ret;
}
int main(){
    n=read();m=read();
    for(R int i=1;i<=m;i++){
        int x=read(),y=read(),ww=read(),www=read();
        add(x,y,ww);add(y,x,www);
    }
    R int ret=work();
    std::printf("%d",ret==0x3f3f3f3f?-1:ret);
    return 0;
}