转自:http://www.cnblogs.com/eric-blog/archive/2012/05/04/2483402.html
输出等差数列由小到大?如果没有符合条件的就输出“NO”。
例如: 输入[1,3,0,5,-1,6] ,输出[-1,1,3,5] 。
要求:时间复杂度,空间复杂度尽量小
解体思路:设随机数为a[],首先排序,
(1)枚举两个数,作为数列中相邻的两个元素Ai,Ai+1,那么可求出等差d=Ai+1-Ai。于是Ai-1=Ai-d,Ai+2=Ai+1+d,因此可以在剩下的数寻找Ai-1,Ai+2是否存在,可以用一个set来维护查找,而数列最长可为n,查找次数最大为n,而查找复杂度为logn,总时间复杂度为O(n^3logn)。复杂度太高,不可取。
(2)动态规划求解。设f[i][j]为以a[i],a[j]结尾的等差数列的最长长度。那么
递归方程为:f[i][j]=max{f[k][i]+1,a[k]-a[i]==a[i]-a[j]},起始条件f[i][j]=1;伪代码如下:
for(int k =0; k <n; k++)
for(int i=k+1;i<n;i++)
for(int j=i+1;j<n;j++)
if(a[k]-a[i] == a[i]-a[j])
f[i][j] = (f[i][j]<f[k][i]+1?f[k][i]+1:f[i][j]);
而,最长长度为即为max{f[i][j]},即以a[i],a[j]结尾,通过一次查找即可求出是那些元素组成的解。复杂度O(n^3)。
(3)枚举等差,将所有差相同的数对(a,b),分别以a,以b作为key,另一个作为值,插入到两个map之中,然后解必然在某一个map之内。于是,枚举每一个map,进行删除操作,比如(a,b)那么下一个值为b+(b-a),前一个值为a-(b-a),分别到两个map中去查找,迭代之。复杂度O(n^2logn)。