【排序-中等】1452. 收藏清单

题目
给你一个数组 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;
    }    
};
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值