状态压缩DP

1494. 并行课程 II

给你一个整数 n 表示某所大学里课程的数目，编号为 1 到 n ，数组 relations 中， relations[i] = [xi, yi] 表示一个先修课的关系，也就是课程 xi 必须在课程 yi 之前上。同时你还有一个整数 k 。
在一个学期中，你 最多 可以同时上 k 门课，前提是这些课的先修课在之前的学期里已经上过了。
请你返回上完所有课最少需要多少个学期。题目保证一定存在一种上完所有课的方式。
提示：

• 1 <= n <= 15
• 1 <= k <= n
• 0 <= relations.length <= n * (n-1) / 2
• relations[i].length == 2
• 1 <= xi, yi <= n
• xi != yi
• 所有先修关系都是不同的，也就是说 relations[i] != relations[j] 。
• 题目输入的图是个有向无环图。

思路

观察到课程数较少，可以使用状态压缩DP来存储每个状态所需的最少学期数。对于每个状态$now$，通过该状态可以得到下次可以修的的课，将其存为一个状态$state$，那么每次可以从$state$中选取至多$k$门课来学习。贪心来讲，修满$k$门课或者将$state$中的所有课都修完就能在最小学期中修完所有课。

代码

class Solution {

    int[] f;

    public int minNumberOfSemesters(int n, int[][] edges, int k) {
        f = new int[1<<n];
        Arrays.fill(f,20);
        for(int[] x : edges) {
            x[0]--;
            x[1]--;
        }
        f[0] = 0;
        for(int i = 0; i < 1 << n; i++) {
            // 标记i状态下，可不以修的课
            boolean[] st = new boolean[n];
            for(int[] x : edges) {
                if(((i >> x[0]) & 1) == 0) {
                    st[x[1]] = true;
                }
            }
            // 标记额外可以修的课
            int state = 0;
            for(int j = 0; j < n; j++) {
                if(!st[j] && ((i>>j) & 1) == 0) {
                    state |= 1 << j;
                }
            }
            dfs(n,k,i,state,0,0);
        }
        return f[(1<<n)-1];
    }

    public void dfs(int n, int k, int now, int state, int add, int u) {
        if(k == 0 || state == 0) {
            f[now | add] = Math.min(f[now|add], f[now] + 1);
            return;
        }
        for(int i = u; i < n; i++) {
            if(((state >> i) & 1) == 1) {
                dfs(n,k-1,now,state - (1<<i), add | (1<<i), i + 1);
            }
        }
    }
}