2021年4月4号,今天清明节。这是2021年4月4号力扣周赛第二道题目,自己还是很菜。第一道字符串分割的题目都搞了半个小时,而这道题写得就过于复杂了。看了大佬的解法醍醐灌顶,参考力扣大佬 @jtjl 的解法重新整理记录一下。
查找用户活跃分钟数
给你用户在 LeetCode 的操作日志,和一个整数 k 。日志用一个二维整数数组 logs 表示,其中每个 logs[i] = [IDi, timei] 表示 ID 为 IDi 的用户在 timei 分钟时执行了某个操作。
多个用户 可以同时执行操作,单个用户可以在同一分钟内执行 多个操作 。
指定用户的 用户活跃分钟数(user active minutes,UAM) 定义为用户对 LeetCode 执行操作的 唯一分钟数 。 即使一分钟内执行多个操作,也只能按一分钟计数。
请你统计用户活跃分钟数的分布情况,统计结果是一个长度为 k 且 下标从 1 开始计数 的数组 answer ,对于每个 j(1 <= j <= k),answer[j] 表示 用户活跃分钟数 等于 j 的用户数。
返回上面描述的答案数组 answer 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/finding-the-users-active-minutes
解题思路
可以构建一个 map,键值对记录 logs 数组的用户ID和其执行了某个操作时的 timei ,这里用set的特性来存放某个用户ID对应的 timei 。遍历二维数组 logs,完成记录,例如 logs = [[0,5],[1,2],[0,2],[0,5],[1,3]], k = 5 时:
对于该 map ,统计用户活跃分钟数等于 j 的用户个数,作为返回结果的数组 answer[j] 。如上,用户活跃分钟数等于 2 的用户个数为 2 。统计时注意,answer 的值是从用户活跃分钟数等于 1 开始记的。
代码
class Solution:
def findingUsersActiveMinutes(self, logs: List[List[int]], k: int) -> List[int]:
mp = defaultdict(set)
ans = [0] * k
for IDi, timesi in logs:
mp[IDi].add(timesi)
for IDi, IDiSet in mp.items():
ans[len(IDiSet) - 1] += 1
return ans
class Solution {
public:
vector<int> findingUsersActiveMinutes(vector<vector<int>>& logs, int k) {
map<int, set<int>> mp;
vector<int> answer(k, 0);
for(auto &it : logs){
mp[it[0]].insert(it[1]);
}
for(auto &it : mp){
answer[it.second.size() - 1]++;
}
return answer;
}
};