特别的排列，状压DP

最新推荐文章于 2024-08-15 23:31:45 发布

仙俊红

最新推荐文章于 2024-08-15 23:31:45 发布

阅读量197

点赞数 1

分类专栏： LeetCode每日一题文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/m0_58809631/article/details/140024520

版权

LeetCode每日一题专栏收录该内容

5 篇文章 0 订阅

订阅专栏

特别的排列
记忆化搜索，dfs(s,i)代表当前可选的集合为s，上一个选择的是nums[i]

typedef long long LL;
class Solution {
public:

  
    int specialPerm(vector<int>& nums) {
  
        int n = nums.size();
        int u = (1 << n) - 1;

        vector<vector<LL>> f(u, vector<LL>(n,-1));
        function<LL(int,int) > dfs = [&](int s, int i)
        {
            if(s == 0)return 1LL;
            LL& res = f[s][i];
            if(res != -1)return res;
            res = 0;
            for(int j = 0; j < n; j ++)
            {
                if( ((s >> j) & 1) && (nums[i] % nums[j] == 0 || nums[j] % nums[i] == 0)    )
                {
                    res += dfs(s ^ (1 << j), j);
                } 
            }
            return res;
        };
        LL res = 0;
        for(int i = 0; i < n; i ++)
        {
            
            res += dfs(u ^ (1 << i),i);
        }
        return res % 1'000'000'007;
    }
};

自底向上，状压DP

f[s][i]表示当前当前可选的集合为s，上一个选择的是nums[i]

typedef long long LL;
class Solution {
public:

  
    int specialPerm(vector<int>& nums) {
  
        int n = nums.size();
        int u = (1 << n) - 1;

        vector<vector<LL>> f(u, vector<LL>(n,0));
        ranges::fill(f[0], 1LL);

        for(int s = 1; s < u; s ++)
        {
            for(int i = 0; i < n; i ++)
            {
                if(s >> i & 1)continue;  //f[s][i],其中s表示已经选的集合，这里表示是即将选nums[i]，但是当前集合没有nums[i]
            
                for(int j = 0; j < n; j ++)
                {
                    if((s >> j & 1) && (nums[j] % nums[i] == 0 || nums[i] % nums[j] == 0 ))
                    {
                        f[s][i] += f[s ^ (1 << j)][j];//这次选择nums[i],上一个选的是nums[j]
                    }
                }
            }
            
        }

       
        LL res = 0;
        for(int i = 0; i < n; i ++)
        {
            
            res += f[u ^ (1 << i)][i];
        }
        return res % 1'000'000'007;
    }
};