NOJ 1526 最長等差子序列

題目連結：http://ac.nbutoj.com/Problem/view.xhtml?id=1526

思路：

把所有公差離散出來（理論上公差個數應該有(n+1)*n/2 , 不過實際最多2000個)

dp[i][j] 表示 公差為j ，以i下標為結尾的子序列 最優解

 

因為dp只需要考慮這次和上一次，所以用滾動數組最佳化記憶體（即公差是N*N，也可以開出來）

#include<stdio.h>#include<iostream>#include<string.h>#include<set>#include<map>using namespace std;#define N 1005int abs(int x){return x>0?x:-x;}int a[N], dp[2][N*2], n; //理論上應該是dp[2][N*N];map<int, int>mymap;set<int>myset;set<int>::iterator p;int main(){int i, j, fir = 0;while(~scanf("%d",&n)){myset.clear();mymap.clear();for(i = 1; i <= n; i++)scanf("%d",&a[i]);for(i = 1; i < n; i++)for(j = i+1; j <= n; j++)myset.insert(abs(a[i]-a[j]));int top = 0;for(p = myset.begin(); p != myset.end(); p++)mymap.insert(make_pair(*p, top++));int cur = 0, ans = 1;for(j=0;j<top;j++)dp[cur][j] = 1;for(i = 2; i <= n; i++){cur ^= 1;for(j=0;j<top;j++)dp[cur][j] = dp[cur^1][j];for(j = i-1; j  ;j--){int cha = mymap.find(abs(a[i]-a[j])) ->second;dp[cur][cha] = max(dp[cur][cha], dp[cur^1][cha]+1);/* 錯誤剪枝，此題資料水可以這樣操作if(dp[cur][cha]>ans) ans = dp[cur][cha];else if(ans >= j+3)break;*/}}if(fir++)puts("");printf("%d",ans);}return 0;}