[51nod-1055] 最长等差数列

题目：

给定大小为 $1 \le N \le 1e4$ 的多重集合 $A$，每个元素$1 \le a_i \le 1e9$。求由集合元素构成的最长等差数列长度。

分析：

开局先排序，剩下全靠猜。一看 $1e4$，就想 $O(N^2)$ 暴力一下，结果写着写着就变成 $O(N^2\log N)$ 了，还感觉挺有道理：设 $dp[i][j]$ 为以 $i$ 结尾公差为 $j$ 的最长等差数列长度。对于一个位置 $i$，枚举 $j < i$，设 $d = a_i - a_j$，则记 $dp[i][d] = max_{j \lt i}\{dp[j][d] + 1\}$，然而这个 $d$ 显然很大，需要离散化一下，于是加一个 $log$ 就 TLE 了。

题解教做人：记 $dp[i][j]$ 为以 $i, j$ 结尾的最长等差数列长度（一看就很靠谱的样子）。对于每个元素 $a_i$，记录双指针 $l, r$，代表 $a_l, a_i, a_j$ 构成了等差数列，则当 $l$ 向左移动时，$l$ 显然应该向右移动。于是就变成枚举 $l$，滑 $r$ ，同时用 $dp[l][i]$ 更新 $dp[i][r]$。这样复杂度就变成了 $O(N^2)$，注意初始化为所有 $dp[i][j] = 2$。

优化：对于一个 $d$，最长等差数列不会超过 $\lfloor\frac{bound}{d}\rfloor + 1, (bound =max\{A\} - min\{A\})$，所以更新 $dp[i][j]$ 时记录一下当前最大值，可以进行剪枝。实测效果良好。至于如何开 $1e4 \times 1e4$ 的数组，那就只有瞎搞了。比如说开 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);
}