Bootstrap

(nice!!!)LeetCode 2959. 关闭分部的可行集合数目(位运算、最短路、Floyd)

2959. 关闭分部的可行集合数目

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
思路:二进制枚举+Floyd
1、n的最大值为10,那么我们直接枚举n个点的状态即可(0表示不选、1表示选)。枚举这里的时间复杂度为0(2^n)。
2、使用Floyd算法来计算每个点之间的距离,时间复杂度为(n^3)
3、最后枚举每两个点之间的距离进行判断即可
细节看注释

class Solution {
public:
    bool check(int n,int maxDistance,vector<bool> sta,int dis[15][15]){
    	
        for(int k=0;k<n;k++){
            if(sta[k]==0) continue;
            for(int i=0;i<n;i++){
                if(sta[i]==0) continue;
                for(int j=0;j<n;j++){
                    if(sta[j]){
                        dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
                    }
                }
            }
        }
        bool flag=1;
        for(int i=0;i<n;i++){
            if(sta[i]==0) continue;
            for(int j=0;j<n;j++){
                if(i!=j&&sta[j]){
                    if(dis[i][j]>100000||dis[i][j]>maxDistance){
                        flag=0;
                        break;
                    }
                }
            }
        }
        return flag;
    }
    int numberOfSets(int n, int maxDistance, vector<vector<int>>& roads) {
    	//用邻接矩阵来记录每个分部之间的距离
        int g[15][15];
        memset(g,0x3f,sizeof g);
        for(int i=0;i<roads.size();i++){
            g[roads[i][0]][roads[i][1]]=min(roads[i][2],g[roads[i][0]][roads[i][1]]);
            g[roads[i][1]][roads[i][0]]=min(roads[i][2],g[roads[i][1]][roads[i][0]]);
        }
        //注意分部到自己的距离为0
        for(int i=0;i<n;i++){
            g[i][i]=0;
        }
        //ans用于记录符合要求的方案数,初始为1,是将所有分布都删掉的情况
        int ans=1;
        //二进制枚举所有可能的情况
        for(int i=1;i<(1<<n);i++){
            vector<bool> sta(n,0);
            //找出当前状态下被保留的分部
            for(int j=0;j<n;j++){
                if(i>>j&1){
                    sta[j]=1;
                }
            }
            //复制一份邻接矩阵给dis,好进行Floyd
            int dis[15][15];
            memcpy(dis,g,sizeof g);
            if(check(n,maxDistance,sta,dis)) ans++;
        }
        return ans;
    }
};
;