文章目录
一、题目
1、题目描述
2、基础框架
- C++ 版本给出的基础框架代码如下:
class Solution {
public:
vector<bool> friendRequests(int n, vector<vector<int>>& restrictions, vector<vector<int>>& requests) {
}
};
3、原题链接
二、解题报告
1、思路分析
(
1
)
(1)
(1) 把整件事情拆解开来:
(
1.1
)
(1.1)
(1.1) restrictions[i] = [xi, yi]
表示这两人不能在一个集合中。我们叫它 黑名单。
(
1.2
)
(1.2)
(1.2) requests[j] = [uj, vj]
询问这两个人在不在一个集合中。如果在一个集合中,返回true
;如果不在一个集合中,分两种情况:能建立连接 和 不能建立连接。
(
2
)
(2)
(2) 如果对于两个不在集合中的人,建立连接以后,导致这两个集合里面的人存在于 restrictions[i]
中,则这两个人无法建立连接。
(
3
)
(3)
(3) 再来捋一下思路:一开始是一些的孤立的点,询问是否是好友,可以采用并查集
O
(
1
)
O(1)
O(1) 判断。如果是好友,返回true
,不改数据;如果不是好友,则遍历其中一个集合中的所有人的restrictions
,查找它所有的黑名单,是不是在另一个集合中,如果都没有,则能够进行合并;否则不能。
2、时间复杂度
最坏时间复杂度 O ( n 2 ) − O ( n 3 ) O(n^2) - O(n^3) O(n2)−O(n3) 。
3、代码详解
class Solution {
#define maxn 1010
int f[maxn];
vector<int> blacklist[maxn];
bool hash[maxn];
void init(int n) {
for(int i = 0; i < n; ++i) {
f[i] = i;
}
}
void unionset_union(int x, int y) {
int fx = unionset_find(x);
int fy = unionset_find(y);
f[fx] = fy;
}
int unionset_find(int x) {
return f[x] == x ? x : ( f[x] = unionset_find(f[x]) );
}
public:
vector<bool> friendRequests(int n, vector<vector<int>>& restrictions, vector<vector<int>>& requests) {
vector<bool> responses;
int i, j, k;
init(n);
for(i = 0; i < n; ++i) {
blacklist[i].clear();
}
for(i = 0; i < restrictions.size(); ++i) {
int u = restrictions[i][0];
int v = restrictions[i][1];
blacklist[u].push_back(v);
blacklist[v].push_back(u);
}
for(i = 0; i < requests.size(); ++i) {
int u = requests[i][0];
int v = requests[i][1];
int uroot = unionset_find(u);
int vroot = unionset_find(v);
if(uroot == vroot) {
responses.push_back(true);
}else {
memset(hash, 0, sizeof(hash));
for(j = 0; j < n; ++j) {
if( vroot == unionset_find(j) ) {
hash[j] = 1;
}
}
bool flag = 0;
for(j = 0; j < n; ++j) {
if( uroot != unionset_find(j) ) {
continue;
}
for(k = 0; k < blacklist[j].size(); ++k) {
int black = blacklist[j][k];
if(hash[black]) {
flag = 1;
break;
}
}
if(flag) {
break;
}
}
if(flag) {
responses.push_back(false);
}else {
unionset_union(u, v);
responses.push_back(true);
}
}
}
return responses;
}
};
三、本题小知识
四、加群须知
相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」。
那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:
如果链接被屏蔽,或者有权限问题,可以私聊作者解决。
大致题集一览:
为了让这件事情变得有趣,以及「 照顾初学者 」,目前题目只开放最简单的算法 「 枚举系列 」 (包括:线性枚举、双指针、前缀和、二分枚举、三分枚举),当有 一半成员刷完 「 枚举系列 」 的所有题以后,会开放下个章节,等这套题全部刷完,你还在群里,那么你就会成为「 夜深人静写算法 」专家团 的一员。
不要小看这个专家团,三年之后,你将会是别人 望尘莫及 的存在。如果要加入,可以联系我,考虑到大家都是学生, 没有「 主要经济来源 」,在你成为神的路上,「 不会索取任何 」。
🔥联系作者,或者扫作者主页二维码加群,加入刷题行列吧🔥