LeetCode - 413. Arithmetic Slices

解题代码:

（1）

classSolution {

public:

    int numberOfArithmeticSlices(vector<int>&A) {

        if(A.size()<3)

            return 0;

        int count=0;

        for(int i=0;i<A.size()-2;i++){

            int dif=A[i+1]-A[i];

            for(int j=i+2;j<A.size();j++){

                if(A[j]-A[j-1]==dif)

                    count++;

                else

                    break;

            }

        }

        return count;

    }

};

 

（2）

classSolution {

public:

    intnumberOfArithmeticSlices(vector<int>& A) {

        if(A.size()<3)

            return 0;

        vector<int> dif(A.size()-1);

        int count=0;

        int b=0;

        dif[0]=A[1]-A[0];

        for(int i=1;i<A.size()-1;i++){

            dif[i]=A[i+1]-A[i];

            if(dif[i]==dif[i-1])

                b++;

            else{

                count+=b*(b+1)/2;

                b=0;

            }

        }

count+=b*(b+1)/2;

        return count;

    }

};

解题思路:

题目要求输出给出的集合的等差数列的个数。首先若集合的元素少于三个，则不可能构成等差数列。然后，从第一个元素开始，计算它与后一个元素的差，然后计算它的后一个元素与后两个元素的差，若与之前一样，则这几个元素可构成等差数列，而若后面的元素依旧能与之构成等差数列，则又可构成另一个等差数列。以此类推，可得出所有的等差数列个数。（见代码1）

另外，我又尝试了另一种算法。先算出数组中相邻元素之间的差值，然后将这些元素作对比，若有连续的相同的元素，则表示他们可以组成等差数列，其中，若有连续的b个相同的差，则代表可以组成b*(b-1)/2个等差数列。（见代码2）