前言
别样的时光做着同样的事情,你是否像我一样秉烛夜读,伏案而作。愿你我不忘初心,励志前行。
煽情结束了,大兄弟,进入正题。
例题
- 1583. count unhappy friends
问题描述
给你一份 n
位朋友的亲近程度列表,其中 n
总是 偶数 。
对每位朋友i
,preferences[i]
包含一份 按亲近程度从高到低排列 的朋友列表。换句话说,排在列表前面的朋友与i
的亲近程度比排在列表后面的朋友更高。每个列表中的朋友均以 0
到 n-1
之间的整数表示。
所有的朋友被分成几对,配对情况以列表 pairs
给出,其中pairs[i] = [xi, yi]
表示 xi
与yi
配对,且yi
与xi
配对。
但是,这样的配对情况可能会是其中部分朋友感到不开心。在x
与y
配对且 u
与v
配对的情况下,如果同时满足下述两个条件,x
就会不开心:
x
与 u
的亲近程度胜过x
与y
,且u
与 x 的亲近程度胜过u
与 v
返回 不开心的朋友的数目 。
示例
示例 1: 输入:n = 4, preferences = [[1, 2, 3], [3, 2, 0], [3, 1, 0], [1, 2, 0]], pairs = [[0, 1], [2, 3]] 输出:2 解释: 朋友 1 不开心,因为: - 1 与 0 配对,但 1 与 3 的亲近程度比 1 与 0 高,且 - 3 与 1 的亲近程度比 3 与 2 高。 朋友 3 不开心,因为: - 3 与 2 配对,但 3 与 1 的亲近程度比 3 与 2 高,且 - 1 与 3 的亲近程度比 1 与 0 高。 朋友 0 和 2 都是开心的。 示例 2: 输入:n = 2, preferences = [[1], [0]], pairs = [[1, 0]] 输出:0 解释:朋友 0 和 1 都开心。 示例 3: 输入:n = 4, preferences = [[1, 3, 2], [2, 3, 0], [1, 3, 0], [0, 2, 1]], pairs = [[1, 3], [0, 2]] 输出:4
题目解析
乍一看,不知道这题目在说啥。换个思路似乎变得有意思多了,想来 leetcode
为了迎合特殊的日子也是煞费苦心,哈哈。
情景一:七夕节当天你和对象出去玩,还是报团的那种。导游为了让团内的气氛更加活跃,于是随机把团内的成员分成了两两一组,很遗憾你和对象不在同一组,这个时候你和你对象肯定都不开心对吧,哈哈。
情景二:还是七夕节当天你和你对象出去玩,场景和上面一样,但是这个时候情况发生了变化,你对象遇到了他/她的初恋,而且被分在了同一组,这个时候你一定不开心,但是你对象开心呀,这个时候你只能成全他们,假装开心,哈哈。
情景三:依然是七夕节当天你和你对象出去玩,场景和上面一样,但是情况和上面两个都不一样,你对象的初恋也在这个团里面,但是没有和他分在同一组。这个时候你会不开心,因为和你对象在同一组的人还不如你,哈哈。
现在是不是感觉明朗多了。
代码示例
package _1582_count_unhappy_friends
import (
"math"
)
func unhappyFriends(n int, preferences [][]int, pairs [][]int) int {
// 初始化亲密度矩阵
relationship := make([][]int, n)
for i := 0; i < n; i++ {
relationship[i] = make([]int, n)
for j := 0; j < n; j++ {
relationship[i][j] = math.MaxInt32
}
}
// 记录各个朋友亲密程度 i 代表各个朋友的编号
for i := 0; i < n; i++ {
for k, v := range preferences[i] {
relationship[i][v] = k
}
}
// 记录配对朋友
forcePairs := make(map[int]int)
for _, pair := range pairs {
forcePairs[pair[0]] = pair[1]
forcePairs[pair[1]] = pair[0]
}
var unhappyMembers int
// 遍历每一组成员,找到他/她爱的人
for _, pair := range pairs {
relation := []int{
relationship[pair[0]][pair[1]], // relationXToY
relationship[pair[1]][pair[0]], // relationYToX
}
// 判断组内的每个成员是否幸福
for XY, memberX := range pair {
for memberU, relationXU := range relationship[memberX] {
flag := false
if relationXU < relation[XY] { // 说明 memberX 没有和爱的人在一起
relationUX := relationship[memberU][memberX]
// 查找 memberX 爱人 memberU 的爱人
for memberV, relationUV := range relationship[memberU] {
if forcePairs[memberU] == memberV && relationUX < relationUV {
// 说明 memberX 的爱人也没有和她/他更爱的人在一起,当然就不幸福咯
unhappyMembers++
flag = true
break
}
// 这里当然说明 memberX 的爱的人的对象,比她/他更好。
// memberX 当然成全她/他咯,哈哈
}
// 确定 memberX 不幸福直接找找下一个不幸福的人
if flag {
break
}
}
}
}
}
return unhappyMembers
}
参考资料
[1] leetcode
[2] https://github.com/unendlichkeiten/LeetcodeWithGo