每日一题 1601最多可达成的换楼请求数目(子集模版)

题目

1601
我们有 n 栋楼,编号从 0 到 n - 1 。每栋楼有若干员工。由于现在是换楼的季节,部分员工想要换一栋楼居住。

给你一个数组 requests ,其中 requests[i] = [fromi, toi] ,表示一个员工请求从编号为 fromi 的楼搬到编号为 toi 的楼。

一开始 所有楼都是满的,所以从请求列表中选出的若干个请求是可行的需要满足 每栋楼员工净变化为 0 。意思是每栋楼 离开 的员工数目 等于 该楼 搬入 的员工数数目。比方说 n = 3 且两个员工要离开楼 0 ,一个员工要离开楼 1 ,一个员工要离开楼 2 ,如果该请求列表可行,应该要有两个员工搬入楼 0 ,一个员工搬入楼 1 ,一个员工搬入楼 2 。

请你从原请求列表中选出若干个请求,使得它们是一个可行的请求列表,并返回所有可行列表中最大请求数目。

示例 1:

输入:n = 5, requests = [[0,1],[1,0],[0,1],[1,2],[2,0],[3,4]]
输出:5
解释:请求列表如下:
从楼 0 离开的员工为 x 和 y ,且他们都想要搬到楼 1 。
从楼 1 离开的员工为 a 和 b ,且他们分别想要搬到楼 2 和 0 。
从楼 2 离开的员工为 z ,且他想要搬到楼 0 。
从楼 3 离开的员工为 c ,且他想要搬到楼 4 。
没有员工从楼 4 离开。
我们可以让 x 和 b 交换他们的楼,以满足他们的请求。
我们可以让 y,a 和 z 三人在三栋楼间交换位置,满足他们的要求。
所以最多可以满足 5 个请求。
示例 2:

输入:n = 3, requests = [[0,0],[1,2],[2,1]]
输出:3
解释:请求列表如下:
从楼 0 离开的员工为 x ,且他想要回到原来的楼 0 。
从楼 1 离开的员工为 y ,且他想要搬到楼 2 。
从楼 2 离开的员工为 z ,且他想要搬到楼 1 。
我们可以满足所有的请求。
示例 3:

输入:n = 4, requests = [[0,3],[3,1],[1,2],[2,0]]
输出:4

题解

class Solution {
    private int max = 0;
    private int[][] requests;
    private int[] buildings;
    private int cnt = 0;
    public int maximumRequests(int n, int[][] requests) {
        buildings = new int[n];
        this.requests = requests;
        dfs(0);
        return max;
    }
    private void dfs(int i) {
        if (check()) {
            max = Math.max(cnt,max);
        }
        if (i == requests.length) {
            return;
        }
        for (int j = i; j < requests.length; j++) {
            buildings[requests[j][0]]--;
            buildings[requests[j][1]]++;
            cnt++;
            dfs(j+1);
            //返回操作
            buildings[requests[j][0]]++;
            buildings[requests[j][1]]--;
            cnt--;
        }
    }
    //判断是否分配合理
    private boolean check() {
        for (int i : buildings) {
            if (i != 0) {
                return false;
            }
        }
        return true;
    }
}

你可能感兴趣的:(深度优先,算法,数据结构,leetcode,java)