集合间最短路 (dij+二进制划分集合)

题意:有n个点,m条边,给出k个点,从k个点中选出两个点使得这两点间的距离最小,求这个最小距离。

题解:把这k个点分成两个集合,按照该点 第 i 位是否为1进行划分,每次划分都要跑一次dij,这样循环跑logn次就能将所有点集情况都遍历到。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#include<vector> 
using namespace std;
#define inf 0x7f7f7f7f
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define mem(a,b) memset(a,b,sizeof(a));
#define lowbit(x)  x&-x;  
#define debugint(name,x) printf("%s: %d\n",name,x);
#define debugstring(name,x) printf("%s: %s\n",name,x);
typedef long long ll;
typedef unsigned long long ull;
const double eps = 1e-6;
const int maxn = 1e5+5;
const int mod = 1e9+7;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

ll dis[505];
int mp[505][505],vis[505],num[505],n,m;
queue<int>q;

void dij(){
    while(!q.empty()) q.pop();
    for(int i = 1; i <= n; i++){
        if(dis[i] == 0) 
            q.push(i);
    }
    mem(vis, 0);
    while(!q.empty()){
        int u = q.front();
        q.pop();
        vis[u] = 0;
        for(int i = 1; i <= n; i++){
            if(mp[u][i] + dis[u] < dis[i]){
                dis[i] = dis[u] + mp[u][i];
                if(!vis[i]){
                    vis[i] = 1;
                    q.push(i);
                }
            }
        }
    }
}
int main() {
    int t;
    cin>>t;
    while(t--){
        int u,v,w;
        scanf("%d%d",&n,&m);
        mem(mp,inf);
        for(int i = 1; i <= m; i++){
            scanf("%d%d%d",&u,&v,&w);
            mp[u][v] = min(mp[u][v],w);
        }
        int k;
        scanf("%d",&k);
        mem(num,0);
        ll ans = 1e18;
        for(int i = 1; i <= k; i++)
            scanf("%d",&num[i]);
        for(int i = 0; i <= 10; i++){
            mem(dis,inf);
            for(int j = 1; j <= k; j++){
                if(num[j]&(1<<i)){  //1
                    dis[num[j]] = 0;
                }
            }
            dij();
            for(int j = 1; j <= k; j++)
                if(dis[num[j]] && !(num[j]&(1<<i))) 
                    ans = min(ans,dis[num[j]]);
            mem(dis,inf);
            for(int j = 1; j <= k; j++){
                if(!(num[j]&(1<<i))){ //0
                    dis[num[j]] = 0;
                }
            }
            dij();
            for(int j = 1; j <= k; j++)
                if(dis[num[j]] && (num[j]&(1<<i))) 
                    ans = min(ans,dis[num[j]]);
        }
        printf("%lld\n",ans);
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值