leetcode 1583. 统计不开心的朋友-Count Unhappy Friends

Problem: 1583. 统计不开心的朋友-Count Unhappy Friends

用哈希表记录preferences数组中每个数字的索引arr，哈希表记录每个人对应的pair中另一个人indexPair

遍历pair，拿到x, y，然后对0-arr[x][y]索引内，找到对应的u和v，然后在0-arr[u][v]内查找x若找到ans++，break

然后交换x和y, 再查找一次的

Code

class Solution { public: int unhappyFriends(int n, vector<vector<int>>& preferences, vector<vector<int>>& pairs) { vector<int> tr(n, -1), indexPair(n); vector<vector<int>> arr(n); int n1 = n - 1, n2 = n/2, x, y, u, v; for(int i = 0; i < n; i++) { tr[i] = -1; for(int j = 0; j < n1; j++) tr[preferences[i][j]] = j; arr[i] = tr; } for(int i = 0; i < n2; i++) { x = pairs[i][0]; y = pairs[i][1]; indexPair[x] = y; indexPair[y] = x; } int ind, ans = 0; bool find; for(int i = 0; i < n2; i++) { x = pairs[i][0]; y = pairs[i][1]; for(int iii = 0; iii < 2; iii++) { if(iii == 1) swap(y, x); find = false; for(int j = 0; j < arr[x][y]; j++) { u = preferences[x][j]; v = indexPair[u]; for(int k = 0; k < arr[u][v]; k++) { if(preferences[u][k] == x) { find = true; ans++; } } if(find) break; } } } return ans; } };