Leetcode-1004 Max Consecutive Ones III(最大连续1的个数 III)

#define pb push_back
#define maxSize 3939
#define _for(i,a,b) for(int i = (a);i < (b);i ++)

class Solution
{
    public:
        int longestOnes(vector<int>& A, int K)
        {
            vector<int> v;
            for(int i = 0;i < A.size();)
            {
                int cnt = 0;
                while(i < A.size() && A[i]==1)
                {
                    cnt ++;
                    i ++;
                }
                v.pb(cnt);
                cnt = 0;
                while(i < A.size() && A[i]==0)
                {
                    cnt ++;
                    i ++;
                }
                v.pb(cnt);
            }
            int rnt = 0;
            for(int i = 0;i < v.size();i += 2)
            {
                int tmpk = K;
                int tmprnt = v[i];
                int j = i;
                while(tmpk)
                {
                    if(tmpk>=v[j+1]&&j+2<v.size())
                    {
                        tmprnt += v[j+2]+v[j+1];
                        tmpk -= v[j+1];
                        j += 2;
                    }
                    else
                    {
                        tmprnt += tmpk;
                        break;
                    }
                    
                }
                rnt = max(rnt,tmprnt);
            }
            int sz = A.size();
            return min(rnt,sz);
        }
};

这题也是乱搞，没有前几周的Ⅱ难，只要对每个连续的数字1块进行遍历，看看以他为头能往后延长多少就行，算是个贪心吧。这么贪的理由是，最后形成的最长1串必定包含某一个原来的数字1块，并且以这一个数字1块开头。