2020滴滴面试算法题：垃圾分类

垃圾分类

最近很多城市都搞起了垃圾分类，已知有一个小区有 n堆垃圾需要丢弃，这些垃圾都打包了，我们并不知道这是什么垃圾，要知道有些垃圾如果丢在一起是会影响环境的。这个小区一共只有两辆垃圾车，我们希望在不影响环境的情况下，每次让垃圾车载走最多的垃圾，但是因为两位司机师傅有矛盾，所以两辆车装的垃圾数目一定要相同，不然其中一位司机师傅就会不开心。
我们为垃圾袋标了号，分别是 1-n，有 m 个约束，每个约束表示为“a b”，意思是第a堆垃圾与第b堆垃圾不能装在一辆垃圾车上。(每堆垃圾最多有两个约束条件)
请问两辆垃圾车一次最多可以带走多少堆垃圾呢?

输入：

输入：
- 输入第一行包含两个正整数 n，m，表示共有 n 堆垃圾，m 个约束条件。
- 接下来有 m 行，每行有两个正整数，a，b，表示第 a 堆垃圾和第 b 堆垃圾不能放在一辆垃圾车上。
输出：
- 输出仅包含一个正整数，表示两辆垃圾车一次最多带走的垃圾的数量。

示例：

输入：
     5 2
     1 4
     3 4
输出：4

提示：

• 1 <= n, m <= 500
• 1 <= a, b <= n

题目解析

把垃圾当作点，限制条件当作边，建一个图。
由于每个垃圾最多有两个限制，那么每个点要么独立存在，要么在一个环上面，要么在一个链上面。

1. 链上的点都可以依次放入两辆垃圾车上。
2. 如果一个环包含偶数个点，那么这个环上的点也都可以依次放入两辆垃圾车上。
3. 如果一个环包含奇数个点，那么必然有一个点是两辆车上都放不了的，我们就要扔掉一个点。
4. 如果最后可以放置的垃圾的个数是奇数，则答案减一，以保证两辆车装的一样多。

因此，借助并查集，找到包含奇数个点的环的数量即可。

PS：如果对并查集不熟悉，这篇文章中有详细介绍：（刷题必备）并查集模板

本题参考代码：

public class Solution {
    static int[] parent;
    static int[] size;

    public static int find(int x) {
        // 路径压缩
        return x==parent[x] ? x : (parent[x]=find(parent[x]));
    }

    public static boolean unite(int x, int y) {
        int p1 = find(x);
        int p2 = find(y);
        if(p1 == p2) {
            return false;
        }
        // 按秩合并
        if(size[p1] < size[p2]) {
            int temp = p1;
            p1 = p2;
            p2 = temp;
        }
        parent[p2] = p1;
        size[p1] += size[p2];
        return true;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();

        // 初始化并查集
        parent = new int[n + 1];
        size = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            parent[i] = i;
            size[i] = 1;
        }
        int ans = n;
        for (int i = 0; i < m; i++) {
            int x = scanner.nextInt();
            int y = scanner.nextInt();
            if(!unite(x, y)) {
                // 在输入的数据不重复的情况下，
                // 如果x和y已经在一个集合中，再合并x、y必然形成一个环
                // 如果环中元素是奇数个，则ans减1
                if(size[find(x)] % 2 == 1) {
                    ans--;
                }
            }
        }
        // ans如果为奇数需要-1，保证两辆车装的一样多
        ans -= (ans&1);
        System.out.println(ans);
    }
}