列表

详情


1949. 坚定的友谊

表名: Friendship

+-------------+------+
| 列名         | 类型 |
+-------------+------+
| user1_id    | int  |
| user2_id    | int  |
+-------------+------+
(user1_id, user2_id) 是这个表的主键(具有唯一值的列的组合)。
这张表的每一行都表示用户 user1_id 和 user2_id 是朋友。
请注意,user1_id < user2_id。

 

如果 x  和 y 为 朋友 且他们 至少 有三个共同的朋友 ,那么 x 和 y 之间的友谊就是 坚定的

写一个解决方案来找到所有的 坚定的友谊

注意,结果表不应该包含重复的行,并且 user1_id < user2_id

以 任何顺序 返回结果表。

返回结果格式如下所示。

 

示例 1:

输入: 
Friendship table:
+----------+----------+
| user1_id | user2_id |
+----------+----------+
| 1        | 2        |
| 1        | 3        |
| 2        | 3        |
| 1        | 4        |
| 2        | 4        |
| 1        | 5        |
| 2        | 5        |
| 1        | 7        |
| 3        | 7        |
| 1        | 6        |
| 3        | 6        |
| 2        | 6        |
+----------+----------+
输出: 
+----------+----------+---------------+
| user1_id | user2_id | common_friend |
+----------+----------+---------------+
| 1        | 2        | 4             |
| 1        | 3        | 3             |
+----------+----------+---------------+
解释: 
用户 1 和 2 有 4 个共同的朋友(3,4,5,和 6)。
用户 1 和 3 有 3 个共同的朋友(2,6,和 7)。
但这里不包括用户 2 和 3 的友谊,因为他们只有两个共同的朋友(1 和 6)。

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
# Write your MySQL query statement below

mysql 解法, 执行用时: 254 ms, 内存消耗: 0 B, 提交时间: 2023-10-15 23:08:23

with a as (
select * from friendship 
union 
select user2_id,user1_id from friendship)

select f.user1_id,f.user2_id,count(*) as common_friend
from friendship f join a a1 
on f.user1_id = a1.user1_id 
join a a2 
on f.user2_id = a2.user1_id 
and a2.user2_id = a1.user2_id 
group by f.user1_id,f.user2_id
having count(*)>=3

mysql 解法, 执行用时: 283 ms, 内存消耗: 0 B, 提交时间: 2023-10-15 23:08:10

# Write your MySQL query statement below
with t as (
    (select user1_id user_id,user2_id friend_id
    from Friendship
    union all
    select user2_id,user1_id
    from Friendship)
    order by 1,2
    )
select t1.user_id user1_id
    ,t2.user_id user2_id
    ,count(distinct t1.friend_id) common_friend 
from t t1,t t2
where (t1.user_id,t2.user_id) in (select * from t)
and t1.friend_id = t2.friend_id
and t1.user_id < t2.user_id
group by 1,2
having count(distinct t1.friend_id) >=3

上一题