[Usaco2004]等差数列

题目大意：约翰发现奶牛经常排成等差数列的号码．他看到五头牛排成这样的序号：“1，4，3，5，7”很容易看出“1，3，5，7”是等差数列。给出N(1≤N≤2000)数字AI..AN(O≤Ai≤10^9)，找出最长的等差数列，输出长度．

数据范围：如题面。

题解：

以为是啥神仙题，结果看见了$1\le N\le 2000$。

可以$N^2$啊.......

考虑$DP$呗，设$f_{(i, j)}$表示第$A_i$个数为等差数列第一项，$A_j$为等差数列第二项的最长等差序列。

显然，我们就需要找到$A_j$后面，离$A_j$最近的等于$2*A_j-A_i$的位置$k$，用$f_{(j, k)} +1$更新$f_{(i, j)}$即可。

这个咋找呢？

我是弄了个$map$，复杂度变成$O(N^2logN)$。

代码：

#include <bits/stdc++.h>

#define N 2010 

using namespace std;

int a[N], f[N][N];

char *p1, *p2, buf[100000];

#define nc() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 100000, stdin), p1 == p2) ? EOF : *p1 ++ )

int rd() {

    int x = 0, f = 1;

    char c = nc();

    while (c < 48) {

        if (c == '-')

            f = -1;

        c = nc();

    }

    while (c > 47) {

        x = (((x << 2) + x) << 1) + (c ^ 48), c = nc();

    }

    return x * f;

}

map<int, int>MP;

int main() {

    int n = rd();

    if (n == 1)

        puts("1"), exit(0);

    for (int i = 1; i <= n; i ++ ) {

        a[i] = rd();

    }

    MP[a[n]] = n;

    for (int i = 1; i < n; i ++ ) {

        f[i][n] = 2;

    }

    for (int j = n - 1; j >= 2; j -- ) {

        for (int i = 1; i < j ; i ++ ) {

            f[i][j] = 2;

            int to = a[j] + a[j] - a[i];

            // int id = MP.count(to);

            // printf("%d %d %d %d %d %d\n", i, j, a[i], a[j], to, id);

            if (MP.count(to)) {

                f[i][j] = max(f[i][j], f[j][MP[to]] + 1);

            }

        }

        MP[a[j]] = j;

    }

    int ans = 0;

    for (int i = 1; i <= n - 1; i ++ ) {

        for (int j = i + 1; j <= n; j ++ ) {

            // printf("%d %d %d\n", i, j, f[i][j]);

            ans = max(ans, f[i][j]);

        }

    }

    cout << ans << endl ;

    return 0;

}

小结：做题看数据范围是很重要的，还有$map$在判断有没有值的时候要用$.count()$，不然会新建点。而且这东西是个$bool$，并不是$[]$的进化版。

巴特西

[bzoj3357][Usaco2004]等差数列_动态规划_贪心

[Usaco2004]等差数列

最新文章

热门文章