摘要:给定未排序的数组,请给出方法找到最长的等差数列。
分析:
题目描述比较简单,我们需要首先搞清楚:题目中要求的等差数列中的数字,是否要和原始数组中的顺序一致。我们在这里分两种情讨论
保证数字的顺序
要求等差数列,
第一、首先可以记录下来数组中任意两个数的差。对于数组A,记录A[j]-A[i],其中 i<j。这里采用hashmap来记录,key就是A[j]-A[i],value是所有差等于的A[j]-A[i]的(i,j)位置对。构造 hashmap的过程时间效率为O(n^2)。
第二、遍历hashmap的每一个value,将位置对按照起始位置进行排序,再进行一次扫描就可以得到最长的等差数列。如下面的数组:{4,7,3,6,9,8}
构造hashmap如下:
-
-4=>(1,2)
-
-1=>(4,5)
-
3=>(0,1)(2,3)(3,4)
上面已经排好序,对于第一个,找到等差数列2,3,4对应数字为3,6,9。第二个、第三个只有一组。方法平均时间复杂度为O(n^2)。
无需保证数字的顺序
因为原来的数组是无序的,因此我们先对数组进行排序,最终的一定是排序之后序列的子序列。然后,我们采用动态规划(dynamic programming)的方法解决这个问题。
假设dp[i][j]表示以A[i]A[j]开始的数列的长度,dp[i][j]如何表示呢?dp[i][j]=dp[j][k]+1,当 A[j]-A[i]=A[k]-A[j],及A[k]+A[i]=2*A[j]。根据dp[i][j]的定义,我们知道dp[x][n-1]=2,也就是 最后一列是2,数列只有A[x]和A[n-1]两个元素。首先,j从n-2,开始向前遍历,对于每一个,找到i和k,满足 A[k]+A[i]=2*A[j],则有dp[i][j]=dp[j][k]+1,若没有,则dp[i][j]就为2.
这里找i和k,有一个小技巧,如下:初始i=j-1,k=j+1,然后分别向两边遍历,如果A[k]+A[i]2*A[j]则i--。完整代码如下:
#include "stdafx.h"
#include <iostream>
using namespace std;
int lengthofLongestDp(int set[],int n)
{
if(n<=2)return n;
const int N=10;
int L[N][N];
int llap=2;
for(int i=0;i<n;++i)
L[i][n-1]=2;
for(int j=n-2;j>=1;--j)
{
int i=j-1,k=j+1;
while(i>=0&&k<=n-1)
{
if(set[i]+set[k]<2*set[j])
k++;
else if(set[i]+set[k]>2*set[j])
{
L[i][j]=2,i--;
}
else
{
L[i][j]=L[j][k]+1;
llap=max(llap,L[i][j]);
i--,k++;
}
}
while(i>=0)
{
L[i][j] =2;
i--;
}
}
return llap;
}
int main()
{
int set[10]={1,2,3,4,5,6,7,88,9,10};
cout<<lengthofLongestDp(set,10);
return 0;
}