LeetCode - 413. Arithmetic Slices

    xiaoxiao2021-03-25  107

    解题代码:

    (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)

    转载请注明原文地址: https://ju.6miu.com/read-24198.html

    最新回复(0)