leetcode 第 44 场双周赛 1733. 需要教语言的最少人数 B

leetcode 第 44 场双周赛  1733. 需要教语言的最少人数 B

https://leetcode-cn.com/problems/minimum-number-of-people-to-teach/

1733. 需要教语言的最少人数

难度中等5

在一个由 m 个用户组成的社交网络里，我们获取到一些用户之间的好友关系。两个用户之间可以相互沟通的条件是他们都掌握同一门语言。

给你一个整数 n ，数组 languages 和数组 friendships ，它们的含义如下：

• 总共有 n 种语言，编号从 1 到 n 。
• languages[i] 是第 i 位用户掌握的语言集合。
• friendships[i] = [u​​​​​​i​​​, v​​​​​​i] 表示 u​​​​​​​​​​​i​​​​​ 和 vi 为好友关系。

你可以选择 一门 语言并教会一些用户，使得所有好友之间都可以相互沟通。请返回你 最少 需要教会多少名用户。

请注意，好友关系没有传递性，也就是说如果 x 和 y 是好友，且 y 和 z 是好友， x 和 z 不一定是好友。

示例 1：

输入：n = 2, languages = [[1],[2],[1,2]], friendships = [[1,2],[1,3],[2,3]]
输出：1
解释：你可以选择教用户 1 第二门语言，也可以选择教用户 2 第一门语言。

示例 2：

输入：n = 3, languages = [[2],[1,3],[1,2],[3]], friendships = [[1,4],[1,2],[3,4],[2,3]]
输出：2
解释：教用户 1 和用户 3 第三门语言，需要教 2 名用户。

提示：

• 2 <= n <= 500
• languages.length == m
• 1 <= m <= 500
• 1 <= languages[i].length <= n
• 1 <= languages[i][j] <= n
• 1 <= u​​​​​​i < v​​​​​​i <= languages.length
• 1 <= friendships.length <= 500
• 所有的好友关系 (u​​​​​i, v​​​​​​i) 都是唯一的。
• languages[i] 中包含的值互不相同。

分析

选一种语言教给其他人，O(n)复杂度，看一下哪种语言好

然后对于每一种选出来的语言，判断最少教给多少人，复杂度O(m)，遍历所有的人

注意提前把已经好的关系删掉，不然要一直重复计算，直接上O(n^3)

这道题比赛的时候没A出来

看了大佬的代码，进行了预处理，然后用数组屏蔽掉了那些无用的

可以适当的进行浪费，保证更优的策略

代码

#include <vector>
vector<vector<int> > len(605);
class Solution {
public:
    int minimumTeachings(int n, vector<vector<int>>& languages, vector<vector<int>>& friendships){
        
        for (int i = 0;  i < len.size(); i++) {
            len[i].resize(600);
        }
        //return len.size();
        for(int i = 0; i < len.size(); i ++){
            for(int j = 0; j < len[i].size(); j ++){
                len[i][j] = 0;
            }
        }
        //return languages[1].size();
        for(int i = 0; i < languages.size(); i ++){
            for(int j = 0; j < languages[i].size(); j ++){
                len[i+1][languages[i][j]] = 1;
            }
        }
        int xx = friendships.size();
        //
        for(int i = 0; i < xx; i ++){
            int a = friendships[i][0];
            int b = friendships[i][1];
            for(int j = 1; j <= n; j ++){
                if(len[a][j] == 1 && len[b][j] == 1){
                    friendships.erase(friendships.begin()+i);
                    i --;
                    xx --;
                    break;
                }
            }
        }
        //return friendships.size();
        /*vector<vector<int>>::iterator it = friendships.begin()+1;
        friendships.erase(it);*/
        int mini = 10000;
        for(int i = 1; i <= n; i ++){
            map<int,int> m;
            m.clear();
            int temp = 0;
            for(int j = 0; j < friendships.size(); j ++){
                int a = friendships[j][0];
                int b = friendships[j][1];
                if(len[a][i]==0 && m[a] == 0){
                    m[a] ++;
                    temp ++;
                }
                if(len[b][i]==0 && m[b] == 0){
                    m[b] ++;
                    temp ++;
                }
            }
            mini = min(mini,temp);
            m.clear();
        }
        return mini;
    }
};