264. 通用子数组个数

264. 通用子数组个数

 

给定一个由 2 或 4 组成的数组。如果一个数组的子数组  (子数组是数组中相邻的一组元素且不能为空)  符合以下条件，则称为“通用”:

1. 2 和 4 被连续分组(如[4, 2],[2, 4],[4, 4, 2, 2],[2, 2, 4, 4],[4, 4, 4, 2, 2, 2]等等)。
2. 子数组中 4 的个数等于子数组中 2 的个数。
3. 相同元素但位置不同的子数组视为不同，如数组[4, 2, 4, 2]中有两个[4, 2]子数组。

你需要返回一个整数值，即给定数组中“通用”子数组的数量。

样例

 

样例 1：

输入样例：array = [4, 4, 2, 2, 4, 2]

输出样例：4

样例解释：匹配这两个条件的  4  个子数组包括：[4, 4, 2, 2]，[4,2]，[2,4]，[4,2]。注意有两个子数组[4,2]，分别在索引 1-2 和 4-5 中。

 

样例 2：

输入样例：array = [4, 4]

输出样例：0

样例解释：数组中没有  2 ，肯定无法满足条件二，所以结果为 0。

注意事项

 

1 ≤ |array| ≤ 10^5

array[i] ∈ (2, 4)

 

public class Solution {

    /**

     * @param array: An given array.

     * @return: Return the number of "universal" subarrays.

     */

    public int subarrays(int[] array) {

        // write your code here.

        if (array.length < 2) return 0;

        int ret = 0;

        int flag2 = 0;

        int flag4 = 0;

        for (int i = 0; i < array.length - 1; i++) {

            if (array[i] == 4) {

                flag4++;

            } else {

                flag2++;

            }

            if (array[i] != array[i + 1]) {

                if (flag2 != 0 && flag4 != 0) {

                    ret = ret + Math.min(flag2, flag4);

                    if (array[i + 1] == 4) {

                        flag4 = 0;

                    } else {

                        flag2 = 0;

                    }

                }

            }

        }

        if (array[array.length - 1] == 4) {

            flag4++;

        } else {

            flag2++;

        }

        if (flag2 != 0 && flag4 != 0) {

            ret = ret + Math.min(flag2, flag4);

        }

        return ret;

    }

}