leedcode周赛 需要交语言的最少人数(两种写法)

在一个由 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^3)将到O(n^2)1.将每个人会什么语言使用unordered_map<int,unordered_set<int>>
记录下来,使用unordered_set可以使查找时时间复杂度降到O(1)。
这步操作时间复杂度O(n^2)2.将朋友关系中原本就不能相互沟通的朋友对筛选出来。
时间复杂度O(n^2)3.遍历每一门语言,再遍历每一个不能沟通朋友对,判断是否会当前语言,
不会则教会该人,人数加一,同时注意使用unordered_set集合记录下
已经教会了哪些人。该步操作同样只需要O(n^2)。

复杂度分析:
时间复杂度:O(n^2)
空间复杂度:O(n^2)


class Solution {
public:
    int minimumTeachings(int n, vector<vector<int>>& languages, vector<vector<int>>& friendships) {
        // 步骤1
        unordered_map<int,unordered_set<int>> memo;
        for (int i = 0;i < languages.size();++i){
            for (int j = 0;j < languages[i].size();++j){
                memo[i + 1].insert(languages[i][j]);
            }
        }
        // 步骤2
        vector<vector<int>> temp;
        for (int i = 0;i < friendships.size();++i){
            int a = friendships[i][0],b = friendships[i][1];
            int flag = 0;
            for (int j = 0;j < languages[a - 1].size();++j){
                if (memo[b].count(languages[a - 1][j])){
                    flag = 1;
                    break;
                }
            }
            if (flag == 0) temp.push_back({a,b});
        }
        if (temp.size() == 0) return 0;
        // 步骤3
        int ans = INT_MAX;
        for (int i = 1;i <= n;++i){
            unordered_set<int> st;
            int count = 0;
            for (int j = 0;j < temp.size();++j){
                int a = temp[j][0],b = temp[j][1];
                if (!memo[a].count(i) && !st.count(a)){
                    ++count;
                    st.insert(a);
                }
                if (!memo[b].count(i) && !st.count(b)){
                    ++count;
                    st.insert(b);
                }
            }
            ans = min(ans,count);
        }
        return ans;
    }
};


//贪心
class Solution {
public:
    int minimumTeachings(int n, vector<vector<int>>& languages, vector<vector<int>>& friendships) {
                  int m = languages.size();
        vector<vector<bool>> g(m + 1, vector<bool>(n + 1));
        for (int i = 0; i < languages.size(); i ++ ) {
            for (auto x: languages[i]) {
                g[i + 1][x] = true;//用户会这种语言就标记true
            }
        }
         set<int> persons;
        for (auto& f: friendships) {
            int x = f[0], y = f[1];
            bool flag = false;
            for (int i = 1; i <= n; i ++ ) //遍历所有语言
                if (g[x][i] && g[y][i]) { //好友之间能交流
                    flag = true;
                    break;
                }
            if (flag) continue;
            persons.insert(x), persons.insert(y);
        }
         int s = 0;
        vector<int> cnt(n + 1);
        for (auto x: persons) {
            for (int i = 1; i <= n; i ++ )
                if (g[x][i]) {
                    cnt[i] ++ ;
                    s = max(s, cnt[i]);
                }
        }

        return persons.size() - s;

    }
};
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 鲸 设计师:meimeiellie 返回首页