题目:
给定大小为 1≤N≤1e4 1 ≤ N ≤ 1 e 4 的多重集合 A A ,每个元素。求由集合元素构成的最长等差数列长度。
分析:
开局先排序,剩下全靠猜。一看 1e4 1 e 4 ,就想 O(N2) O ( N 2 ) 暴力一下,结果写着写着就变成 O(N2logN) O ( N 2 log N ) 了,还感觉挺有道理:设 dp[i][j] d p [ i ] [ j ] 为以 i i 结尾公差为 的最长等差数列长度。对于一个位置 i i ,枚举 ,设 d=ai−aj d = a i − a j ,则记 dp[i][d]=maxj<i{dp[j][d]+1} d p [ i ] [ d ] = m a x j < i { d p [ j ] [ d ] + 1 } ,然而这个 d d 显然很大,需要离散化一下,于是加一个 就 TLE 了。
题解教做人:记 dp[i][j] d p [ i ] [ j ] 为以 i,j i , j 结尾的最长等差数列长度(一看就很靠谱的样子)。对于每个元素 ai a i ,记录双指针 l,r l , r ,代表 al,ai,aj a l , a i , a j 构成了等差数列,则当 l l 向左移动时, 显然应该向右移动。于是就变成枚举 l l ,滑 ,同时用 dp[l][i] d p [ l ] [ i ] 更新 dp[i][r] d p [ i ] [ r ] 。这样复杂度就变成了 O(N2) O ( N 2 ) ,注意初始化为所有 dp[i][j]=2 d p [ i ] [ j ] = 2 。
优化:对于一个 d d ,最长等差数列不会超过 ,所以更新 dp[i][j] d p [ i ] [ j ] 时记录一下当前最大值,可以进行剪枝。实测效果良好。至于如何开 1e4×1e4 1 e 4 × 1 e 4 的数组,那就只有瞎搞了。比如说开 short, 比如说开一半等。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N_Max = 1e4 + 10;
int N;
short dp[N_Max][N_Max];
int arr[N_Max];
inline short Max(short a, short b) {
return a > b ? a : b;
}
int main() {
scanf("%d", &N);
for (int i = 1; i <= N; i++) {
scanf("%d", arr + i);
}
sort(arr + 1, arr + N + 1);
int ans = 0;
for (int i = 1; i < N; i++) {
for (int j = i + 1; j <= N; j++) {
dp[i][j] = 2;
ans = Max(ans, dp[i][j]);
}
}
int bound = arr[N] - arr[1];
for (int i = 2; i < N; i++) {
int l = i - 1, r = i + 1;
while (l >= 1 && r <= N) {
if (arr[l] + arr[r] > 2 * arr[i]) l--;
else if (arr[l] + arr[r] < 2 * arr[i]) r++;
else {
dp[i][r] = Max(dp[i][r], dp[l][i] + 1);
ans = Max(ans, dp[i][r]);
if (bound / (arr[r] - arr[i]) <= ans) break;
if (l == 1) r++;
else l--;
}
}
}
printf("%d\n", ans);
}