题解 | #牧场奶牛集合区域#

知识点

数组，模拟，STL（vector）

思路

不断更新区间片段的末尾，若出现不连续区间，则将当前区间的首尾更新至答案中，并且更新新的区间的首，详解看代码注释。

注意两个情况： ①初始化队头，末尾区间在循环结束后需要再更新到答案中一次。 ②对一个数单独成区间的情况，判断返回时，临时区间数组的size是否为1，若是，则说明当前区间只有一个数，再插入一个自己作为尾即可。

代码c++

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param groups int整型vector 
     * @param n int整型 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > findGatheringAreas(vector<int>& groups, int n) {
        // write code here
        vector<vector<int>>ans;   //最终返回的答案

        vector<int>t;//临时存储区间首尾

        t.push_back(groups[0]);//初始化第一个队头
        for(int i=1;i<n;i++)
        {  
            if(groups[i]-groups[i-1]==1)//判断区间连续性
            {
               if(t.size()>=2) t.pop_back();//有两个数时，更新末尾区间
                t.push_back(groups[i]);
            }
            else {    

             if(t.size()==1)t.push_back(t[0]);     //单独一个数的情况
             ans.push_back(t);           //更新答案
             t.clear();//清空临时存答案的向量
             t.push_back(groups[i]);//初始化新的队头
            }
        }
        if(t.size()==1)t.push_back(t[0]);//处理末尾一个数的情况
        ans.push_back(t);//更新答案
        return ans;
    }
};