Leetcode_1583_统计不开心的朋友_模拟

最新推荐文章于 2022-03-03 21:43:57 发布

原创最新推荐文章于 2022-03-03 21:43:57 发布 · 132 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

daily_algorithm 同时被 2 个专栏收录

230 篇文章

订阅专栏

模拟

22 篇文章

订阅专栏

博主分享了如何通过合并循环和使用数组简化`unhappyFriends`函数，避免使用哈希表索引查找，以提高代码效率并直接AC。主要讨论了合并偏好顺序和配对比较的优化策略。

写了一坨屎山，我自己都看不下去了orz
很明显x和u的循环是可以合并的，y和v的循环也是可以合并的
既然编号都是有序的，那直接用数组就完事了，存个der的hash表，还得浪费时间去找索引，懒得改了，ac就完事了

class Solution {
    public int unhappyFriends(int n, int[][] preferences, int[][] pairs) {
        int ans = 0;
        int len = pairs.length;
        Map<Integer, Integer>[] map = new Map[n];
        for (int i = 0; i < n; i++) {
            map[i] = new HashMap<Integer, Integer>();
            for (int j = 0; j < n-1; j++) {
                map[i].put(preferences[i][j], n - j);
            }
        }

        final HashMap<Integer, Boolean> happy = new HashMap<>();

        for (int i = 0; i < len; i++) {
            int x = pairs[i][0];
            int y = pairs[i][1];
            for (int j = i + 1; j < len; j++) {
                int u = pairs[j][0];
                int v = pairs[j][1];
                if(map[x].get(u) > map[x].get(y) && map[u].get(x) > map[u].get(v) || map[x].get(v) > map[x].get(y) && map[v].get(x) > map[v].get(u)) {
                    happy.put(x, true);
                }
                if(map[y].get(u) > map[y].get(x) && map[u].get(y) > map[u].get(v) || map[y].get(v) > map[y].get(x) && map[v].get(y) > map[v].get(u)) {
                    happy.put(y, true);
                }
                if(map[u].get(x) > map[u].get(v) && map[x].get(u) > map[x].get(y) || map[u].get(y) > map[u].get(v) && map[y].get(u) > map[y].get(x)) {
                    happy.put(u, true);
                }
                if(map[v].get(x) > map[v].get(u) && map[x].get(v) > map[x].get(y) || map[v].get(y) > map[v].get(u) && map[y].get(v) > map[y].get(x)) {
                    happy.put(v, true);
                }
            }
        }

        return happy.keySet().size();
    }
}