leetcode 1131 获取你好友已观看的视频

在这里插入图片描述

class Solution {
public:
    vector<string> watchedVideosByFriends(vector<vector<string>>& watchedVideos, vector<vector<int>>& friends, int id, int level) {
        vector<bool> vis(friends.size(),false);
        map<string,int> m;
        vector<string> res;
        int floor=0;
        queue<int> q;
        q.push(id);
        vis[id]=true;
        while(!q.empty()){
            if(floor==level){
                while(!q.empty()){
                    int cur = q.front();
                    q.pop();
                    for(int i=0;i<watchedVideos[cur].size();i++){
                        if(!m[watchedVideos[cur][i]])
                            m[watchedVideos[cur][i]]=1;
                        else
                            m[watchedVideos[cur][i]]++;
                    }  
                }
                break;
            }
            
            int size = q.size();
            for(int j=0;j<size;j++){
                id = q.front();q.pop();
                for(int i=0;i<friends[id].size();i++){
                    if(!vis[friends[id][i]]){
                        q.push(friends[id][i]);
                        vis[friends[id][i]]=true;                 
                    }
                }
            }
            floor++;
        }
        
        vector<pair<string, int>> vec(m.begin(),m.end());
        sort(vec.begin(), vec.end(), cmp);
        map<string, int>::iterator it;
        
        for (int i = 0; i < vec.size(); ++i){
	        res.push_back(vec[i].first);
	    }
        return res;
    }
    static bool cmp(const pair<string, int>& a, const pair<string, int>& b) {
	        return a.second < b.second||(a.second==b.second && a.first < b.first);
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值