【题目】
给你一个数组 favoriteCompanies ,其中 favoriteCompanies[i] 是第 i 名用户收藏的公司清单(下标从 0 开始)。
请找出不是其他任何人收藏的公司清单的子集的收藏清单,并返回该清单下标。下标需要按升序排列。
【示例 1】
输入:favoriteCompanies = [[“leetcode”,“google”,“facebook”],[“google”,“microsoft”],[“google”,“facebook”],[“google”],[“amazon”]]
输出:[0,1,4]
解释:
favoriteCompanies[2]=[“google”,“facebook”] 是 favoriteCompanies[0]=[“leetcode”,“google”,“facebook”] 的子集。
favoriteCompanies[3]=[“google”] 是 favoriteCompanies[0]=[“leetcode”,“google”,“facebook”] 和 favoriteCompanies[1]=[“google”,“microsoft”] 的子集。
其余的收藏清单均不是其他任何人收藏的公司清单的子集,因此,答案为 [0,1,4] 。
【示例 2】
输入:favoriteCompanies = [[“leetcode”,“google”,“facebook”],[“leetcode”,“amazon”],[“facebook”,“google”]]
输出:[0,1]
解释:favoriteCompanies[2]=[“facebook”,“google”] 是 favoriteCompanies[0]=[“leetcode”,“google”,“facebook”] 的子集,因此,答案为 [0,1] 。
【示例 3】
输入:favoriteCompanies = [[“leetcode”],[“google”],[“facebook”],[“amazon”]]
输出:[0,1,2,3]
【提示】
1 <= favoriteCompanies.length <= 100
1 <= favoriteCompanies[i].length <= 500
1 <= favoriteCompanies[i][j].length <= 20
favoriteCompanies[i] 中的所有字符串 各不相同 。
用户收藏的公司清单也 各不相同 ,也就是说,即便我们按字母顺序排序每个清单, favoriteCompanies[i] != favoriteCompanies[j] 仍然成立。
所有字符串仅包含小写英文字母。
【代码】
【Python】
执行用时:
1740 ms, 在所有 Python3 提交中击败了5.71%的用户
内存消耗:
22.3 MB, 在所有 Python3 提交中击败了80.95%的用户
class Solution:
def peopleIndexes(self, favoriteCompanies: List[List[str]]) -> List[int]:
temp=set()
for index,favoriteCompanie in enumerate(favoriteCompanies):
for i in range(len(favoriteCompanies)):
if i==index:
continue
if len(set(favoriteCompanie)&set(favoriteCompanies[i]))==len(set(favoriteCompanie)):
temp.add(index)
break
rs=list(set([i for i in range(len(favoriteCompanies))])-temp)
rs.sort()
return rs
【CPP】
执行用时:
180 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:
43 MB, 在所有 C++ 提交中击败了84.68%的用户
class Solution {
public:
vector<int> peopleIndexes(vector<vector<string>>& favoriteCompanies) {
for (auto& f : favoriteCompanies) {
sort(f.begin(), f.end());
}
map<int, vector<int>> mapSizeIndex;
for (int i = 0; i < favoriteCompanies.size(); i++) {
mapSizeIndex[favoriteCompanies[i].size()].push_back(i);
}
vector<int> ans;
for (auto it = mapSizeIndex.rbegin(); it != mapSizeIndex.rend(); it++) {
for (auto& p : it->second) {
if (it != mapSizeIndex.rbegin() && checkIncludes(favoriteCompanies, ans, p)) continue;
ans.push_back(p);
}
}
sort(ans.begin(), ans.end());
return ans;
}
bool checkIncludes(vector<vector<string>>& fc, vector<int>& ans, int p) {
for (auto i : ans) {
if (includes(fc[i].begin(), fc[i].end(), fc[p].begin(), fc[p].end())) return true;
}
return false;
}
};