[luogu] P5858「SWTR-03」Golden Sword

题目详情
题目分析：
这里的第一种方法有一种剪枝的思想。这个方法是由上一层枚举下一层，在每一次枚举中如果遇到比下一层这个数还要大，说明这个数已经被上一层前面数枚举过了，而且这个数字比当前这个数还要大，所以就没有继续枚举的必要率了，直接退出。

#include <cstdio>
#include <algorithm>
#define MIN -0x3f3f3f3f3f3f
using namespace std;
long long n, w, s, a[5555], dp[5555][5555], ans = MIN;
int main()
{
    scanf("%lld%lld%lld", &n, &w, &s);
    for (int i = 1; i <= n; i++) scanf("%lld", a + i);
    for (int i = 0; i <= n; i++)
        for (int j = 0; j <= w; j++)
            dp[i][j] = MIN;
    dp[0][0] = 0;
    for (int i = 0; i < n; i++)
        for (int j = 0; j <= w && j <= i; j++)
            for (int k = j + 1; k >= 1 && k >= j - s + 1; k--)
            {
                if (dp[i + 1][k] > dp[i][j] + a[i + 1] * k)
                    break;
                dp[i + 1][k] = dp[i][j] + a[i + 1] * k;
            }
    for (int i = 1; i <= w; i++)
        ans = max(ans, dp[n][i]);
    printf("%lld\n", ans);
}

这个方法用到了滚动数组和单调队列。

#include <cstdio>
#include <algorithm>
using namespace std;
long long q[5555], n, w, s, a[5555], dp[2][5555], row = 1, ans = -1e15;
int main()
{
    scanf("%lld%lld%lld", &n, &w, &s);
    for (int i = 1; i <= n; i++) scanf("%lld", a + i);
    for (int i = 0; i <= 1; i++)
        for (int j = 0; j <= w; j++)
            dp[i][j] = -1e15;
    dp[0][0] = 0;
    for (int i = 1; i <= n; i++)
    {
        for (int j = 0; j <= w; j++)
            dp[row][j] = -1e15;
        int h = 1, t = 1;
        q[h] = w;
        for (int j = w; j >= 1; j--)
        {
            while (h <= t && q[h] > min(w, j + s - 1)) h++;
            while (h <= t && dp[row ^ 1][q[t]] < dp[row ^ 1][j - 1]) t--;//单点队列。
            q[++t] = j - 1;
            dp[row][j] = dp[row ^ 1][q[h]] + j * a[i];
        }
        row ^= 1;//变成另外一行
    }
    for (int i = 0; i <= w; i++)
        ans = max(ans, dp[n & 1][i]);//n奇偶不同最后一行的位置就不同。
    printf("%lld\n", ans);
}