D.Maximum Sum of Products(DP、暴力、双指针)

七万米深海

已于 2022-04-04 10:14:52 修改

阅读量637

点赞数 1

于 2022-04-03 01:22:37 首次发布

本文链接：https://blog.csdn.net/qq_51583015/article/details/123931873

版权

动态规划算法

题目：D. Maximum Sum of Products

watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5LiD5LiH57Gz5rex5rW3,size_20,color_FFFFFF,t_70,g_se,x_16

题意：

给出长度为n的数组a和数组b，最多可以翻转一次（也可以不翻转）a数组的某一个子区间，使 $\sum_{i=1}^{n}a_{i} * b_{i}$ 最大，数据范围： $1 \leqslant n \leqslant5000$ 。

思路：

暴力所有区间为 $gif.latex?n^3$ 复杂度，会超时，发现小区间包含在大区间中，计算增量时指针不必每次从头开始，考虑状态转移，定义 $gif.latex?f[i][j]$ 为翻转 $gif.latex?[i, j]$ 这个区间带来的增量，区间 $gif.latex?[i, j]$ 由区间 $gif.latex?[i+1, j-1]$ 转移而来，则有:

$gif.latex?f[i][j] = f[i+1][j-1] + a[i]b[j] + a[j]b[i] - a[i]b[i] - a[j]b[j]$

边界： $gif.latex?f[i][i] = 0$ ， $gif.latex?f[i][i-1] = 0$ ，左端点i从n-1开始枚举，右端点从i+1开始枚举，两重循环计算出增量最大的区间，最大增量加上翻转前的 $\sum_{i=1}^{n}a_{i} * b_{i}$ 即为答案。

代码：

#include <iostream>

using namespace std;

typedef long long LL;
const int N = 5010;
LL a[N], b[N];
LL f[N][N];
int n;

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
        cin >> a[i];
    LL sum = 0;
    for(int i = 1; i <= n; i++)
    {
        cin >> b[i];
        sum += a[i] * b[i];
    }
    for(int i = 1; i <= n; i++)
        f[i][i] = 0, f[i][i-1] = 0;
    LL add = 0;
    for(int i = n-1; i >= 1; i--)
        for(int j = i+1; j <= n; j++)
        {
            f[i][j] = f[i+1][j-1] + a[i]*b[j] + a[j]*b[i] - a[i]*b[i] - a[j]*b[j];
            add = max(add, f[i][j]);
        }
    cout << sum + add << endl;
    return 0;
}
//f[i][j] = f[i+1][j-1] + a[i]*b[j] + a[j]*b[i] - a[i]*b[i] - a[j]*b[j]