牛客2020跨年场 B.牛牛想起飞

牛客2020跨年场 B.牛牛想起飞

题目链接

题目描述

给定一个长度为 $n$ 的序列 $a$ 和一个长度为 $n$ 的序列 $b$，对于序列中的每一个数字 $a_i$，牛牛都可以进行以下操作：
(1) 将数字 $a_i$ 从 $a_i$ 变成 $a_i+b_i$；
(2) 将数字 $a_i$ 从 $a_i$ 变成 $a_i-b_i$。
牛牛很懒，所以每个数字只能最多进行一次操作，甚至不操作。现在给出一个数字 y ，牛牛想要知道在做合理的操作后能得到模 y 意义下的最大序列和是多少。
换句话说，设 $a_1^{\prime },a_2^{\prime },a_3^{\prime }\dots a_n^{\prime }$ 是最终得到的 数组的样子，你要使得 $(a_1^{\prime }+a_2^{\prime }+a_3^{\prime }\cdots +a_n^{\prime })mody$ 结果最大化。
做出这道题他就能起飞，你能帮帮他吗？

输入描述:

第一行输入 $n,y(1\le n\le 10^5,1\le y\le 100)$
第二行输入 $n$ 个整数 $a_i(0\le a_i\le 10^5)$
第三行输入 $n$ 个整数 $b_i(0\le b_i\le 100)$

输出描述:

输出能得到模 y 意义下的最大序列和

示例1

输入

3 100
1 2 3
1 1 1

输出

9

动态规划~
设 $dp[i][j]$ 前 $i$ 个数通过操作后取模可以得到值 $j$，则有状态转移方程：

$ifdp[i-1][j]then\{\begin{array}{l}dp[i][(j+a[i])\%y]=1\\ dp[i][(j+a[i]+b[i])\%y]=1\\ dp[i][(j+a[i]-b[i])\%y]=1\end{array}$
AC代码如下：

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int N = 1e5 + 5;
bool dp[N][105];
int n, a[N], b[N], y;

int mod(int x) {
    return x < 0 ? (y - (-x) % y) % y : x % y;
}

int main() {
    cin >> n >> y;
    for (int i = 1; i <= n; i++) cin >> a[i];
    for (int i = 1; i <= n; i++) cin >> b[i];
    dp[1][mod(a[1])] = 1;
    dp[1][mod(a[1] + b[1])] = 1;
    dp[1][mod(a[1] - b[1])] = 1;
    for (int i = 1; i <= n; i++) {
        for (int j = 0; j <= y; j++) {
            if (dp[i - 1][j]) {
                dp[i][mod(j + a[i])] = 1;
                dp[i][mod(j + a[i] + b[i])] = 1;
                dp[i][mod(j + a[i] - b[i])] = 1;
            }
        }
    }
    for (int i = y - 1; i >= 0; i--) {
        if (dp[n][i]) {
            cout << i;
            return 0;
        }
    }
    return 0;
}