最长等差数列_算法题10 最长等差序列问题

最新推荐文章于 2023-04-27 22:25:06 发布

weixin_39797324

最新推荐文章于 2023-04-27 22:25:06 发布

阅读量518

点赞数

文章标签：最长等差数列

本文链接：https://blog.youkuaiyun.com/weixin_39797324/article/details/111736532

版权

本文介绍了如何解决给定数组中找出最长等差数列的问题。通过动态规划的方法，分别讨论了有序数组和无序数组的情况，提供了不同等差值的最长子序列求解思路，并给出了优化后的算法实现，降低了时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给定一个大小为n的数组，要求写出一个算法，求其最长的等差数列的子序列

分析

该题需要分几种情况考虑。

1. 原数组是有序的，所要求的的子序列可以不连续。

对于数组arr[],不同的等差值d=1,2,3,4,5```(arr[max]-arr[min])可以求出不同的最长等差数列，然后在这些等差数列中求出最长的那个即可我们首先转化为求一个数组的固定等差值的最长等差子序列。如数组1,2,4,6,8,9，求等差值2的最长等差子序列为2,4,6,8

1.1 固定等差值的最长子序列

求符合条件的最长子序列可以用动态规划来做，设dis[i]记录数组arr[]的第1-i个元素子数组的最长等差子序列长度，状态转移方程式：

dis[i]=1;

dis[i]=dis[k]+1, k是数组第1-i个元素中，与元素i等差为d并且距离i最近的元素(k

代码

1 int RegularArithmeticSeq(int arr[],int len,intd)2 {3 if (arr==NULL||len<1)4 throw std::exception("Invalid input.");5

6 int* dis=new int[len+1]();7

8 int re_len=0,re_index=0;9 for (int i=1;i<=len;i++)10 {11 dis[i]=1;12 for (int k=i;k>0;k--)13 {14 int cur_d=arr[i-1]-arr[k-1];15

16 if (cur_d==d&&dis[i]<=dis[k])17 {18 dis[i]=dis[k]+1;19 }20

21 if (re_len<=dis[i])22 {23 re_len=dis[i];24 re_index=i;25 }26

27 if (cur_d>d)28 break;29 }30 }31

32 int out=arr[re_index-1];33 for (int n=0;n

40 delete[] dis;41

42 returnre_len;43 }

上述代码的时间复杂度为O(n*n),时间复杂度为O(n);

对于寻找1-i中最后一个和i等差的元素k,除了顺序遍历，还可以用二分查找法进行一点优化，查找的时间复杂度为O(logn)

1 int find_k(int arr[],int n,intd)2 {3 int small=0,big=n-1,k=0,sum=0;4 while (small<=big)5 {6 k=small+(big-small)/2;7 sum=arr[k]+d;8 if (sumarr[n-1])12 {13 big=k-1;14 }else

15 {16 if (k<=big&&(arr[k+1]+d)!=sum)17 {18 returnk;19 }else

20 {21 small=k+1;22 }23 }24 }25

26 return n-1;27 }

优化代码

1 int RegularArithmeticSeq2(int arr[],int len,int d,int&re_idx)2 {3 if (arr==NULL||len<1)4 throw std::exception("Invalid input.");5

6 int* dis=new int[len]();7

8 int* dif=new int[len]();9 for (int i=0;i

14 int re_len=0,re_index=0;15 for (int i=0;i

29 int out=arr[re_index];30 re_idx=re_index+1;31 for (int n=0;n

38 delete[] dis;39 delete[] dif;40

41 returnre_len;42 }

上述过程中时间复杂度降为O(n*logn)

1.2 非固定等差值的最长子序列

在解决了固定等差值得最长子序列后，就可以着手求非固定等差值的最长子序列了。等差值d的范围是1,2···(arr[max]-arr[min])，循环着m个等差值，然后求出最长的等差子序列即可

时间复杂度(n*logn*m),m<=n-1;空间复杂度O(n*m)

代码

1 int ArithmeticSeq(int arr[],intlen)2 {3 if (arr==NULL||len<1)4 throw std::exception("Invalid input.");5

6 int max_d=arr[len-1]-arr[0];7 int re_len=0,re_d=0,re_index=0;8

9 for (int d=1;d<=max_d;d++)10 {11 int idx=0;12 int dlen=RegularArithmeticSeq2(arr,len,d,idx);13 if(re_len<=dlen)14 {15 re_len=dlen;16 re_d=d;17 re_index=idx;18 }19 }20

21 int out=arr[re_index-1];22 for (int n=0;n

29 returnre_len;30 }

也有另外一种解法：设dis[i][d]为数组中第1到i个元素中等差值为d的最长子序列的长度(序列可以不连续)，则动态规划式;

dis[i][d]=1,默认等1，即在1-i个元素中等差值为d的最长子序列长度至少是1(i元素本身)

dis[i][d]=dis[k][d]+1, k为1-i个元素中最后一个与i元素的等差d的元素

代码

1 int ArithmeticSeq(int arr[],intlen)2 {3 if (arr==NULL||len<1)4 throw std::exception("Invalid input.");5

6 int max_d=arr[len-1]-arr[0];7 int** dis=new int*[len+1];8 for (int i=0;i<=len;i++)9 {10 dis[i]=new int[max_d+1]();11 dis[i][0]=1;12 }13

14 int re_len=0,re_d=0,re_index=0;15 for (int i=1;i<=len;i++)16 {17 for (int d=1;d<=max_d;d++)18 {19 dis[i][d]=1;20 for (int k=1;k

28 if (re_len<=dis[i][d])29 {30 re_len=dis[i][d];31 re_d=d;32 re_index=i;33 }34 }35

36 }37 }38

39 int out=arr[re_index-1];40 for (int n=0;n