leetcode413 Aritmetic Slices Java

Description

A sequence of number is called arithmetic if it consists of at least three elements and if the difference between any two consecutive elements is the same.

For example, these are arithmetic sequence:

1, 3, 5, 7, 9
7, 7, 7, 7
3, -1, -5, -9
The following sequence is not arithmetic.

1, 1, 2, 5, 7

A zero-indexed array A consisting of N numbers is given. A slice of that array is any pair of integers (P, Q) such that 0 <= P < Q < N.

A slice (P, Q) of array A is called arithmetic if the sequence:
A[P], A[p + 1], …, A[Q - 1], A[Q] is arithmetic. In particular, this means that P + 1 < Q.

The function should return the number of arithmetic slices in the array A.

Example:

A = [1, 2, 3, 4]

return: 3, for 3 arithmetic slices in A: [1, 2, 3], [2, 3, 4] and [1, 2, 3, 4] itself.

解法：

我感觉题目描述的不是太清楚。

1、A不一定是一个等差数列

2、等差数列必须在A中连续

也就是说如果A={1,3,5,6,8}，那么返回结果就是1。{1,3,5}

A={1,3,5,7,6,8}，那么返回结果就是3。{1,3,5}，{3,5,7}，{1,3,5,7}

public int numberOfArithmeticSlices(int[] A) {
        if(A.length <3) return 0;
        int result = 0;
        int curr = 0;
        for(int i=2; i<A.length; i++) {
            if(A[i-1] - A[i-2] == A[i] - A[i-1]) {
                curr ++;
                result += curr;
            }else {
                curr = 0;
            }
        }
        return result;
    }