Codeforces 255C Almost Arithmetical Progression DP

点击打开链接

题意:n个数b[i],定义序列a[i]=a[i-1]+(-1)^(i+1) *q (i>2,q为常量)
n<=4e3,b[i]<=1e6,问b中,子序列满足定义的最长长度?

合法的序列形式为 x,x+q,x,x+q,x.. .
x,x+q 下一个出现的x 一定为第一次出现在x+q之后的x(显然更优)
//怎么快速找到下一个x? b[i]<=1e6,用vector保存每个b[i]的位置,则对v[x] 二分找到第一个下标>j的x

不用二分...从后往前推时,用pos记录J之后第一个和b[i]相等的位置 
则设状态dp[i][j] 以i,j开头的最长子序列 dp[i][j]=dp[j][pos]+1
复杂度O(N^2)

#include <bits/stdc++.h>using namespace std;typedef long long ll;typedef pair<int,int> ii;const int inf=1e9;const int N=2e6+20;const int M=4e3+20;int b[N],n;vector<int> v[N];int dp[M][M];int main(){while(cin>>n){for(int i=0;i<=1e6;i++)v[i].clear();for(int i=1;i<=n;i++)scanf("%d",&b[i]),v[b[i]].push_back(i);int ans=0;if(n==1){cout<<n<<endl;continue;}memset(dp,0,sizeof(dp));for(int i=n;i>=1;i--){int pos=-1;for(int j=n;j>=i+1;j--){if(pos==-1)dp[i][j]=2;elsedp[i][j]=dp[j][pos]+1;//posΪjÖ®ºóµÚÒ»¸ö³öÏÖµÄb[i]if(b[j]==b[i])pos=j; ans=max(ans,dp[i][j]);}}cout<<ans<<endl;}return 0;}