LeetCode 485. Max Consecutive Ones. C++

485. Max Consecutive Ones

题目

Given a binary array, find the maximum number of consecutive 1s in this array.

Input: [1,1,0,1,1,1]
Output: 3
Explanation: The first two digits or the last three digits are consecutive 1s.
The maximum number of consecutive 1s is 3.

解法

很简单的一题，遍历数组，num用来计数，ans用来更新最大连续数目。遇到0元素，就将当前的计数num与之前保存的最大值ans比较一下，然后num清零准备下一次计数。
考虑到最后可能不会遇到0就结束了，所以返回的是num和ans的最大值。比如[1 1 0 1 1 1]，最长的连续段在最后三位，遍历结束时，num=3，ans=2。

class Solution {
public:
    int findMaxConsecutiveOnes(vector<int>& nums) {
       int num=0,ans=0;
       for(int i=0;i<nums.size();i++){
           if(nums[i]==1) num++;
           else{
               ans=max(ans,num);
               num=0;
           }       
       }
       return max(ans,num);
    }
};