Clarke and problem HDU - 5464(dp 类似背包转移)

做一只大熊猫

于 2021-06-08 20:23:43 发布

阅读量96

点赞数 1

分类专栏：线性 DP

本文链接：https://blog.csdn.net/qq_34261446/article/details/117715168

版权

动态规划数组模运算状态转移方程组合计数

关键词由CSDN通过智能技术生成

线性 DP 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目链接

题意

给我们 n 个数的，我们可以从中选择任意个数把这些数的值加起来，要求这个数是 m 的倍数，问我们组成的数字是的倍数是 m 的方案数，
注意一个也不选组成的数是 0 也算一种合法方案。
n <=1000, p<=1000.

思路

一看到组成的数是 m 的倍数就要明白我们可以在 dp 的状态中用一位去枚举余数，
其实 dp 方程很容易就可以猜出来：dp [i][j] 表示前 i 个数从中选择一些数字的和是 % m 的余数是 j 的方案数，那么答案就是 dp [n][0] 了，
那么状态转移方程为：
1. 如果第 i 个数我们不选择：dp [i][j] += dp [i - 1][j]
2. 如果选择第 i 个数：dp [i][(j + a [i]) % m] += dp [i][j]

代码

#include <bits/stdc++.h>
using namespace std;
#define db  double
#define ll  long long
#define sc  scanf
#define pr  printf
#define fi  first
#define se  second
#define pb  push_back
#define m_p make_pair
#define Pir pair<int, int>
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
/*==========ACMer===========*/
const int N = 1005;
const int mod = 1e9 + 7;
int n, p, a[N];
int dp[N][N];


int main()
{
    int T; sc("%d", &T);
    while (T --)
    {
        sc("%d %d", &n, &p);
        for (int i = 1; i <= n; i ++) sc("%d", &a[i]);
        memset(dp, 0, sizeof dp);
        dp[0][0] = 1;

        for (int i = 1; i <= n; i ++)
        {
            for (int j = 0; j < p; j ++)
            {
                if (dp[i - 1][j] == 0) continue;
                int k = (j + a[i] % p + p) % p;
                dp[i][k] = (dp[i][k] + dp[i - 1][j]) % mod;
                dp[i][j] = (dp[i][j] + dp[i - 1][j]) % mod; 
            }
        }
        pr("%d\n", dp[n][0]);
    }

    return 0;
}