leetcode1004+Dp 注意剪枝

https://leetcode.com/problems/max-consecutive-ones-iii/

//cnt_0[i]来表示数组A中前i个数中有多少个0
//若cnt_0[i] - cnt_0[j] > K，那么更小的j值肯定也不满足条件，所以在遍历的时候，
//只要维护一个递增的指针，不满足条件就增加即可，不用从头遍历
class Solution {
public:
    int longestOnes(vector<int>& A, int K) {
        int len = A.size();
        int cnt_0[len+1];
        memset(cnt_0, 0, sizeof(cnt_0));
        int ans=0, j=0;
        for(int i=0; i<len; i++){
            if(A[i]==0){
                cnt_0[i+1]= cnt_0[i]+1;
                while (cnt_0[i+1]-cnt_0[j]>K) {//优化了 cnt_0[j]以前的更不可能了
                    j+=1;
                }
            }
            else{
                cnt_0[i+1] = cnt_0[i];
            }
            ans  = max(ans, i-j+1);
        }
        return ans;
    }
};