[BZOJ2069][POI2004]ZAW:最短路+二进制拆分

分析:

本题求的是从1点出发再回到1点不重复经过点和边的最短路。
可以将1点删去,然后对所有与1点相连的点集进行二进制拆分,然后跑多源最短路即可。

代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
const int MAXN=5005;
const int MAXM=10005;
int n,m,lin[MAXN][3],tot,dis[MAXN],ecnt,head[MAXN];
bool vis[MAXN];
struct Edge{
    int to,nxt,w;
}e[MAXM<<1];
inline void add_edge(int bg,int ed,int val){
    ecnt++;
    e[ecnt].to=ed;
    e[ecnt].nxt=head[bg];
    e[ecnt].w=val;
    head[bg]=ecnt;
}
struct Pair{
    int pos,dis;
    friend bool operator > (Pair x,Pair y){
        return x.dis>y.dis;
    }
};
inline Pair Mp(int x,int y){
    Pair ret;ret.pos=x,ret.dis=y;return ret;
}
priority_queue<Pair,vector<Pair>,greater<Pair> > q;
void dijkstra(){
    while(!q.empty()){
        int u=q.top().pos,v=q.top().dis;q.pop();
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u];i;i=e[i].nxt){
            int ver=e[i].to;
            if(dis[ver]>v+e[i].w){
                dis[ver]=v+e[i].w;
                q.push(Mp(ver,dis[ver]));
            }
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,w,ww;
        scanf("%d%d%d%d",&u,&v,&w,&ww);
        if(u==1) lin[++tot][0]=v,lin[tot][1]=w,lin[tot][2]=ww;
        else if(v==1) lin[++tot][0]=u,lin[tot][1]=ww,lin[tot][2]=w;
        else{
            add_edge(u,v,w);
            add_edge(v,u,ww);
        }
    }
    int ans=1e9;
    for(int i=0;i<=16;i++){
        memset(dis,0x3f,sizeof dis);
        memset(vis,0,sizeof vis);
        for(int j=1;j<=tot;j++)
            if((lin[j][0]>>i)&1) q.push(Mp(lin[j][0],lin[j][1])),dis[lin[j][0]]=lin[j][1];
        dijkstra();
        for(int j=1;j<=tot;j++)
            if(!((lin[j][0]>>i)&1)) ans=min(ans,dis[lin[j][0]]+lin[j][2]);
        memset(dis,0x3f,sizeof dis);
        memset(vis,0,sizeof vis);
        for(int j=1;j<=tot;j++)
            if(!((lin[j][0]>>i)&1)) q.push(Mp(lin[j][0],lin[j][1])),dis[lin[j][0]]=lin[j][1];
        dijkstra();
        for(int j=1;j<=tot;j++)
            if((lin[j][0]>>i)&1) ans=min(ans,dis[lin[j][0]]+lin[j][2]);
    }
    printf("%d\n",ans);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ErkkiErkko/p/9581999.html
今日推荐