1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
get-watched-videos-by-your-friends.cpp 1.34 KB
一键复制 编辑 原始数据 按行查看 历史
// Time: O(n + vlogv), v is the number of the level videos
// Space: O(w)
class Solution {
public:
vector<string> watchedVideosByFriends(vector<vector<string>>& watchedVideos, vector<vector<int>>& friends, int id, int level) {
vector<int> curr_level = {id};
unordered_set<int> lookup = {id};
for (int i = 0; i < level; ++i) {
vector<int> new_level;
for (const auto& i : curr_level) {
for (const auto& j : friends[i]) {
if (lookup.count(j)) {
continue;
}
lookup.emplace(j);
new_level.emplace_back(j);
}
}
curr_level = move(new_level);
}
unordered_map<string, int> count;
for (const auto& i : curr_level) {
for (const auto& v : watchedVideos[i]) {
++count[v];
}
}
vector<string> result;
transform(count.cbegin(), count.cend(), back_inserter(result),
[](const auto& x) {
return x.first;
});
sort(result.begin(), result.end(),
[&count](const auto& a, const auto& b) {
return make_pair(count[a], a) < make_pair(count[b], b);
});
return result;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助