3531:判断整除

总时间限制: 1000ms          内存限制: 65536kB

描述

一个给定的正整数序列，在每个数之前都插入+号或-号后计算它们的和。比如序列：1、2、4共有8种可能的序列：
(+1) + (+2) + (+4) = 7
(+1) + (+2) + (-4) = -1
(+1) + (-2) + (+4) = 3
(+1) + (-2) + (-4) = -5
(-1) + (+2) + (+4) = 5
(-1) + (+2) + (-4) = -3
(-1) + (-2) + (+4) = 1
(-1) + (-2) + (-4) = -7
所有结果中至少有一个可被整数k整除，我们则称此正整数序列可被k整除。例如上述序列可以被3、5、7整除，而不能被2、4、6、8……整除。注意：0、-3、-6、-9……都可以认为是3的倍数。

输入

输入的第一行包含两个数：N（2 < N < 10000）和k(2 < k< 100)，其中N代表一共有N个数，k代表被除数。第二行给出序列中的N个整数，这些整数的取值范围都0到10000之间（可能重复）。

输出

如果此正整数序列可被k整除，则输出YES，否则输出NO。（注意：都是大写字母）

样例输入

3 2
1 2 4

样例输出

NO

 解题思路

每次添加的数字可能是正数，可能是负数，这样构成一个数字序列。

如果最后一个输入的数字是x，前面的数字加和为s

添加x后加和能被k整除，那么有( s + x ) % k = 0 即如果s % k = ( k − x % k ) % k ，那么添加x后加和能被k整除。
添加-x后能被k整除，那么有( s − x ) % k = 0 ， 即如果s % k = x % k 那么添加-x后加和能被k整除。
如果以上两种情况中有一种成立，那么就可以得到能被k整除的加和。
dp[i][j]为是否存在数字序列方案前i个数字的加和除k余j，存在为true；不存在为false。
dp[0][0]为前0个数字的加和是否除k余0，是的。因此dp[0][0] = true。
要求的结果为：前n个数字加和是否可以除k余0，即dp[n][0]

因此可以得出状态转移方程：
dp[i][j] = dp[i-1][j+k-v%k] || dp[i-1][(j+v)%k]
其中v为第i个数字。

#include <bits/stdc++.h>
using namespace std;
bool dp[10005][105]; 
int a[10005];
int main()
{
	int n, k, v;
	cin >> n >> k;
	dp[0][0] = true;
	for(int i = 1; i <= n; ++i)
		cin >> a[i];
	for(int i = 1; i <= n; ++i)
		for(int j = 0; j < k; ++j)
			dp[i][j] = dp[i-1][(k+j-a[i]%k)%k] || dp[i-1][(j+a[i])%k]; 
    cout << (dp[n][0] ? "YES" : "NO");
	return 0;
}