【Asia Yokohama Regional Contest 2018】Arithmetic Progressions

题目大意:给定 N(1

题解:直接暴力有 \(O(n^3)\) 的算法,即:枚举等差数列的前两个值,再暴力枚举后面的值进行匹配即可,不过这样做直接去世。。
考虑 \(dp[i][j]\) 表示以第 i 个数为数列倒数第二位,第 j 个数为等差数列中的最后一位的最长序列的长度,则:\(dp[i][j]=max\{dp[l][i]+1,a[i]-a[l]=a[j]-a[i]\&\&0。不过这样还是要枚举 l 进行转移,考虑序列是有序的,对于外层枚举的 i 来说,j 的枚举过程中 i 是不变的,且 a[j] 的值是单调递增的,可以利用单调性,令 l 不断减小,即可得到答案,均摊复杂度为 \(O(n^2)\)

代码如下

#include 
using namespace std;
const int maxn=5010;

int n,a[maxn],dp[maxn][maxn],ans;

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++){
        int l=i-1;
        for(int j=i+1;j<=n;j++){
            dp[i][j]=2;
            while(l>=1&&a[j]-a[i]>a[i]-a[l])--l;
            if(l>=1&&a[j]-a[i]==a[i]-a[l])dp[i][j]=max(dp[i][j],dp[l][i]+1);
            ans=max(ans,dp[i][j]);
        }
    }
    printf("%d\n",ans);
    return 0;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10632593.html

你可能感兴趣的:(数据结构与算法)