Leetcode C++ 《第175场周赛-3》1348. 推文计数

Leetcode C++ 《第175场周赛-3》1348. 推文计数

1. 题目
   请你实现一个能够支持以下两种方法的推文计数类 TweetCounts：

2. recordTweet(string tweetName, int time)

   记录推文发布情况：用户 tweetName 在 time（以 秒 为单位）时刻发布了一条推文。

3. getTweetCountsPerFrequency(string freq, string tweetName, int startTime, int endTime)

   返回从开始时间 startTime（以 秒 为单位）到结束时间 endTime（以 秒 为单位）内，每 分 minute，时 hour 或者 日 day （取决于 freq）内指定用户 tweetName 发布的推文总数。
   freq 的值始终为 分 minute，时 hour 或者 日 day 之一，表示获取指定用户 tweetName 发布推文次数的时间间隔。
   第一个时间间隔始终从 startTime 开始，因此时间间隔为 [startTime, startTime + delta1>, [startTime + delta1, startTime + delta2>, [startTime + delta2, startTime + delta3>, … , [startTime + deltai, min(startTime + delta*(i+1), endTime + 1)>，其中 i 和 delta（取决于 freq）都是非负整数。

示例：

输入：
[“TweetCounts”,“recordTweet”,“recordTweet”,“recordTweet”,“getTweetCountsPerFrequency”,“getTweetCountsPerFrequency”,“recordTweet”,“getTweetCountsPerFrequency”]
[[],[“tweet3”,0],[“tweet3”,60],[“tweet3”,10],[“minute”,“tweet3”,0,59],[“minute”,“tweet3”,0,60],[“tweet3”,120],[“hour”,“tweet3”,0,210]]

输出：
[null,null,null,null,[2],[2,1],null,[4]]

解释：
TweetCounts tweetCounts = new TweetCounts();
tweetCounts.recordTweet(“tweet3”, 0);
tweetCounts.recordTweet(“tweet3”, 60);
tweetCounts.recordTweet(“tweet3”, 10); // “tweet3” 发布推文的时间分别是 0, 10 和 60 。
tweetCounts.getTweetCountsPerFrequency(“minute”, “tweet3”, 0, 59); // 返回 [2]。统计频率是每分钟（60 秒），因此只有一个有效时间间隔 [0,60> - > 2 条推文。
tweetCounts.getTweetCountsPerFrequency(“minute”, “tweet3”, 0, 60); // 返回 [2,1]。统计频率是每分钟（60 秒），因此有两个有效时间间隔 1) [0,60> - > 2 条推文，和 2) [60,61> - > 1 条推文。
tweetCounts.recordTweet(“tweet3”, 120); // “tweet3” 发布推文的时间分别是 0, 10, 60 和 120 。
tweetCounts.getTweetCountsPerFrequency(“hour”, “tweet3”, 0, 210); // 返回 [4]。统计频率是每小时（3600 秒），因此只有一个有效时间间隔 [0,211> - > 4 条推文。

提示：

同时考虑 recordTweet 和 getTweetCountsPerFrequency，最多有 10000 次操作。
0 <= time, startTime, endTime <= 10^9
0 <= endTime - startTime <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/tweet-counts-per-frequency
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 思路

• 这个题目逻辑不难，但是我掉进了各种类型的二分查找的坑里。比如找大于等于某值的index最小的值，比如找小于等于某值的index最大的值。总有各式各样的数组越界问题。代码其实也不长，一定要严谨。
• 理解题意也很重要！
  • 每个人的发布时间用vector记录，所有人的发布记录用map记录。
  • 当更新某个人的发布时间时：可以通过map找到其记录，然后通过二分查找找到记录应该插入的位置。
  • 获取某个人在某个阶段的间隔发布次数，通过二分查找到这个阶段的发布情况，然后根据间隔进行统计

3. 代码

class TweetCounts {
public:
    map<string, vector<long long>> records;
    TweetCounts() {
        
    }
    void recordTweet(string tweetName, long long time) {
        //cout << "recordTweet start" << endl;
        //插入排序，维护records
        vector<long long> oneRecord;
        map<string, vector<long long>>::iterator it = records.find(tweetName);
        if (it == records.end()){
            oneRecord.push_back(time);
            records.insert(make_pair(tweetName, oneRecord));
            return;
        }
        oneRecord = it->second;
        bool flag = false;
        //插入排序，其实可以通过logn找到这个位置，而不是n

        int insertIndex = binaryFindFirstMoreEqualNum(oneRecord, time);
        if (insertIndex == -1)
            it->second.insert(it->second.end(), time);  
        else
            it->second.insert(it->second.begin()+insertIndex, time);
        /*for (int i = 0; i < oneRecord.size(); i++) {
            if (oneRecord[i] >= time) {
                it->second.insert(it->second.begin()+i, time);
                flag = true;
                break;
            }
        }
        if (flag == false)
            it->second.insert(it->second.end(), time);    */
    }
    
    vector<int> getTweetCountsPerFrequency(string freq, string tweetName, long long startTime, long long endTime) {
        //
        //cout << "getTweetCountsPerFrequency start" << endl;
        int addTime;
        vector<int> res;
        if (freq == "minute") {
            addTime = 60;
        } else if (freq == "hour") {
            addTime = 3600;
        } else if (freq == "day") {
            addTime = 86400;
        } else {
            return res;
        }
        
        map<string, vector<long long>>::iterator it = records.find(tweetName);
        if (it == records.end())
            return res;
        vector<long long> oneRecord = it->second;
        int startIndex = binaryFindFirstMoreEqualNum(oneRecord, startTime);
        //cout << tweetName << " " << startIndex << endl;
        int endIndex = binaryFindLastLessEqualNum(oneRecord, endTime);
        //cout << endIndex << endl;
        int i = startIndex;
        for (int j = startTime; j <= endTime; j=j+addTime) {
            //[j, j+addTime-1]
            int oneSum = 0;
            if (startIndex==-1 || endIndex==-1 || startIndex > endIndex) {
                oneSum = 0;
            } else {
                for (i; i <= endIndex; i++) {
                    if (oneRecord[i] < (j+addTime))
                        oneSum++;
                    else
                        break;
                }
            }
            
            res.push_back(oneSum);
        }
        return res;
    }
        
    int binaryFindFirstMoreEqualNum(vector<long long> nums, long long target)
    {
        //cout << "first more than: "  << target << endl;
        int left = 0;
        int right = nums.size() - 1;
        while(left < right)
        {
            int middle = (left + right) / 2;
            if(nums[middle] > target)
                right = middle;  //保留大于n的下标以防这是第一个
            else   //将小于改为小于等于
                left = middle + 1;
        }
        while (left > 0 && nums[left-1] == target)
            left--;
        //cout << left << endl;
        if(nums[left] >= target)
            return left;
        else
            return -1;
    }
    
    int binaryFindLastLessEqualNum(vector<long long> nums, long long target)
    {
        int x = binaryFindFirstMoreEqualNum(nums, target);
        //cout << x << " less" << endl;
        if (x == -1) {
            x = nums.size()-1;
        }
        while(x >= 0 && nums[x] >target)
            x--;
        if (x >=0 && nums[x] <= target)
            return x;
        else
            return -1;
    }
};

/**
 * Your TweetCounts object will be instantiated and called as such:
 * TweetCounts* obj = new TweetCounts();
 * obj->recordTweet(tweetName,time);
 * vector<int> param_2 = obj->getTweetCountsPerFrequency(freq,tweetName,startTime,endTime);
 */


/*
["TweetCounts","recordTweet","recordTweet","recordTweet","getTweetCountsPerFrequency","getTweetCountsPerFrequency","recordTweet","getTweetCountsPerFrequency"]
[[],["tweet3",0],["tweet3",60],["tweet",10],["minute","tweet3",0,59],["minute","tweet3",0,60],["tweet3",120],["hour","tweet3",0,210]]
*/