LeetCode 413. Arithmetic Slices

可能是本人太水,写的算法实在低效,leetcode都说我是Submission Result: Time Limit Exceeded

没办法,为了吸取教训,以后看的时候能以此警戒自己。。


题目描述就不写了,官网自己找找就好了。

———————————————————————————————————————————

正在学python,所以就用python写了。

class Solution(object):
    def numberOfArithmeticSlices(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        A_dis = []
        num = 0
        for m in range(0, len(A) - 1):
            A_dis.append(A[m + 1] - A[m])
        for n in range(0, len(A_dis)):
            for i in range(n, len(A_dis)):
                if (i - n) == 1 and A_dis[i] == A_dis[n]:
                    num += 1
                elif (i - n) > 1 and A_dis[i] == A_dis[n]:
                    a = A_dis[n:i]
                    c_num = 0
                    for j in a:
                        if j == a[0]:
                            c_num += 1 
                    if c_num == len(a):
                        num += 1
        return num

答题一看很明显能看出来我特么好菜,是很菜。
但是有一颗学习的心,于是在博客上一个博主的思想,

改写之后的代码如下,不得不说,厉害!
class Solution(object):
    def numberOfArithmeticSlices(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        num = 0
        add = 0
        for i in range(2, len(A)):
            if (A[i-1] - A[i]) == (A[i-2] - A[i-1]):
                add += 1
                num += add
            else:
                add = 0
        return num
改进的代码的思想基本是通过观察出等差序列在数目递增过程中前一序列和后一序列的差的递增性,以此作为最终累加之后的等差序列的个数的这个规律,推广到非等差序列数组上。
受教了。
继续努力。。



猜你喜欢

转载自blog.csdn.net/sinat_24648637/article/details/75451811