ICPC Yokohama 2018 B - Arithmetic Progressions

题意：

在序列中找到最多能构成等差数列的数

想法：

排序后 构造 j i k 三个从小到大的指针，每次大循环 i不变，改动j和k
如果是等差数列的话，a[i]-a[j]==a[k]-a[i]
j k 初始化为i的左边一个和右边一个
如果 a[j]+a[k]==2*a[i] 就说明j i k构成等差数列, dp[j][i]=dp[i][k]+1;k++同时j–
如果 a[j]+a[k]<2*a[i] 就说明a[k]还不够大,k++
反之 a[j]+a[k]>2*a[i] 就说明a[j]还不够小，j–
dp数组含义：以a[j]和a[i]作为开头的等差数列的最大长度，任何两个数都能构成长度为2的等差数列，故初始化为2
dp[j][i]=dp[i][k]+1: 能构成a[j] a[i] a[k] 时，显然dp[j][i]=dp[i][k]加一个a[j] 也就是+1

AC代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;

int n,ans=2;
int a[5005],dp[5005][5005];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++)dp[i][j]=2;
    }//初始化
    //小 j i k 大 
    for(int i=n-1;i>1;i--){
        int j=i-1,k=i+1;
        while(j>=1&&k<=n){
            if(a[j]+a[k]==2*a[i]){//等差中项
                dp[j][i]=dp[i][k]+1;
                ans=max(ans,dp[j][i]);
                j--;
                k++;
            }
            else if(a[j]+a[k]<2*a[i]){//a[k]不够大
                k++;
            }
            else {//a[j]不够小 
                j--;
            }
        }
    }
    printf("%d\n",ans);
}