leetcode 第 44 场双周赛 1733. 需要教语言的最少人数 B
https://leetcode-cn.com/problems/minimum-number-of-people-to-teach/
1733. 需要教语言的最少人数
难度中等
在一个由 m
个用户组成的社交网络里,我们获取到一些用户之间的好友关系。两个用户之间可以相互沟通的条件是他们都掌握同一门语言。
给你一个整数 n
,数组 languages
和数组 friendships
,它们的含义如下:
- 总共有
n
种语言,编号从1
到n
。 languages[i]
是第i
位用户掌握的语言集合。friendships[i] = [ui, vi]
表示ui
和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 <= ui < vi <= languages.length
1 <= friendships.length <= 500
- 所有的好友关系
(ui, vi)
都是唯一的。 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; } };
作者:kidgzz
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。