其他分享
首页 > 其他分享> > NC14701 取数游戏2

NC14701 取数游戏2

作者:互联网

题目链接

题目

题目描述

给定两个长度为n的整数列A和B,每次你可以从A数列的左端或右端取走一个数。假设第i次取走的数为ax,则第i次取走的数的价值vi=bi⋅ax,现在希望你求出∑vi的最大值。

输入描述

第一行一个数T,表示有T组数据。
对于每组数据,第一行一个整数n,
接下来两行分别给出A数列与B数列。

输出描述

每一组数据输出一行,最大的∑vi。

示例1

输入

2
2
1 1000
2 1
5
1 3 5 2 4
1 2 3 4 5

输出

2001
52

说明

对于第二个样例,
第一次从左边取走a1,v1=a1⋅b1=1,
第二次从左边取走a2,v2=a2⋅b2=6,
第三次从右边取走a5,v3=a5⋅b3=12,
第四次从右边取走a4,v4=a4⋅b4=8,
第五次取走剩下的a3,v5=a3⋅b5=25。
总价值∑vi=1+6+12+8+25=52

备注

\(T≤10\)
\(1≤n≤10^3\)
\(1≤a_i,b_i≤10^3\)

题解

知识点:区间dp。

这类题有个很显然的特征,即每次只能选剩余的左右两端的数,可以考虑区间dp逆推整个过程,从只剩一个数作为终点逆推回原来的数列,然后取其中过程中的最大值即可。

设 \(dp[i][j]\) 为用区间 \([i,j]\) 里的数作为第 \([n-(j-i),n]\) 个数时能够到达的最大值。有转移方程:

\[dp[i][j] = \max (dp[i + 1][j] + a[i] \cdot b[n - l + 1], dp[i][j - 1] + a[j] \cdot b[n - l + 1]) \]

表示选左端点或者右端点作为第 \(n-l+1\) 个数。

时间复杂度 \(O(n^2)\)

空间复杂度 \(O(n^2)\)

代码

#include <bits/stdc++.h>
#define ll long long

using namespace std;

int a[1007], b[1007], dp[1007][1007];

bool solve() {
    int n;
    cin >> n;
    for (int i = 1;i <= n;i++) cin >> a[i];
    for (int i = 1;i <= n;i++) cin >> b[i];
    for (int i = 1;i <= n;i++) dp[i][i] = a[i] * b[n];
    for (int l = 2;l <= n;l++) {
        for (int i = 1, j = l;j <= n;i++, j++) {
            dp[i][j] = max(dp[i + 1][j] + a[i] * b[n - l + 1], dp[i][j - 1] + a[j] * b[n - l + 1]);
        }
    }
    cout << dp[1][n] << '\n';
    return true;
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t = 1;
    cin >> t;
    while (t--) {
        if (!solve()) cout << -1 << '\n';
    }
    return 0;
}

标签:取走,游戏,int,vi,取数,NC14701,1007,dp,数列
来源: https://www.cnblogs.com/BlankYang/p/16586827.html