动态规划:等差子数组

    给定一个整数数组,计算其等差子数组的个数(至少有算个元素)。如对于数组[1, 2, 3, 4],其等差子数组有[1, 2, 3], [2, 3, 4] 和 [1, 2, 3, 4]。

    分析:只需遍历一次,把等差数列的长度分别计算出来。如果长度为3,其中包含的子数组个数为1,长度为4,包含 2 + 1 个,长度为5,包含 3 + 2 + 1 个,依此类推。

  public int numberOfArithmeticSlices(int[] A) {
        int index = 0;
        int sum = 0;
        while(index < A.length) {
            int l = 2;
            while((index + l < A.length) && A[index + l] - A[index + l - 1] == A[index + 1] - A[index]) l++;
            if(l >= 3)
                for(int i = l - 2; i > 0; i--) 
                    sum += i;
            index += l - 1; //A[index + l]- A[index + l - 1]与前面的差不相等,下一步从A[index + l - 1]开始计算    
            }              
        return sum;
    }


猜你喜欢

转载自blog.csdn.net/xiezongsheng1990/article/details/79998497