Maximum Sum of Products(区间DP)

2022-08-11 09:34:07 浏览数 (1)

题目链接

题意

给你两个长度为n的数列a和b,要求你最多翻转一次a序列的一段连续的区间,使得displaystyle sum_{i = 1} ^ n a_i*b_i 最大。

分析

首先暴力方法是先枚举区间长度再枚举起点,整个过程是 O(n^2) 的,然后就是如何O(1)求出翻转后的值,先去找了一下每个翻转策略之间有没有冗余部分,发现没有啥可以优化的,想不出怎么快速求值,于是否掉了这个方法。开始思考DP,感觉是个区间DP,按照固有思维,将DP的属性设置为某个区间的最优解,但明显不对啊,因为只能翻转一次,只有一种情况,哪里来的最优解。然后就…卡题了。看了一眼题解,oh,wssb。就是区间DP,而且就是用一开始想的暴力找,O(1)算的思路,计算的过程用DP使得每次计算可以O(1)完成。

具体方法是:

定义 dp[i][j] 为区间( i , j )翻转后与翻转前的 displaystyle sum_{k = i} ^ j a_k*b_k 差值,于是 :

dp[i][j] = dp[i 1][j-1] a[i] * b[j] a[j] * b[i] – a[i]*b[i] – a[j] * b[j]

首先算出原数组的 displaystyle sum_{i = 1} ^ n a_i*b_i ,然后记录所有翻转情况和原始数据的差值,然后再找到差值的最大值,两者相加就是答案。

代码

代码语言:javascript复制
#include <bits/stdc  .h>
#define LL long long
using namespace std;
const int maxn = 1e5 10;
const int inf = 0x3f3f3f3f;
const double PI = acos(-1.0);
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
LL a[5555],b[5555];
LL dp[5555][5555];
int main(int argc, char const *argv[]) {
    int n;
    cin >> n;
    for(int i = 1; i <= n; i  ) scanf("%lld",a i);
    for(int i = 1; i <= n; i  ) scanf("%lld",b i);
    LL ans = 0, plus = 0;
    for(int i = 1; i <= n; i  ) {
        ans  = a[i] * b[i];
    }
    for(int len = 2; len <= n; len   ) {
        for(int i = 1; i   len - 1 <= n; i  ) {
            int j = i   len - 1;
            dp[i][j] = dp[i 1][j-1]   a[i] * b[j]   a[j] * b[i] - a[i]*b[i] - a[j] * b[j];
            plus = max(plus,dp[i][j]);
        }
    }
    cout << plus   ans <<endl;
    return 0;
}

0 人点赞