LeetCode 413.等差数列划分

爷,独闯天下 提交于 2020-03-11 01:07:03

如果[1,2,3,4,5,6]是一个等差数列,则

以A[2]结尾的等差数列 [1,2,3] 以A[3]结尾的等差数列 [1,2,3,4],[2,3,4] 以A[4]结尾的等差数列
[1,2,3,4,5],[2,3,4,5],[3,4,5]
以A[5]结尾的等差数列 [1,2,3,4,5,6],[2,3,4,5,6],[3,4,5,6],[4,5,6]

f[i]A[i]为结尾的等差数列的个数。
答案为f[]数组的总和

class Solution {
    public int numberOfArithmeticSlices(int[] A) {
        int n = A.length;
        if(n < 3)
            return 0;
        int []f = new int[n];
        int res=0;
        for(int i = 2;i < n;i++){
            if(A[i]-A[i-1]==A[i-1]-A[i-2])
             f[i]=f[i-1]+1;
             res += f[i];
        }
        return res;
    }
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!