首页 > 代码库 > 【OpenJudge3531】【背包DP】【膜法交配律】判断整除
【OpenJudge3531】【背包DP】【膜法交配律】判断整除
判断整除
总时间限制: 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
【Solution】
首先,膜法交配律,其实就是模法分配律,即(a+b+c)%k=(a%k+b%k+c%k)%k。所以预处理输入数据全部模k。
dp[i][j]表示到第i个数模k是否有可能等于j。转移方程为dp[i+1][(j+data[i]-100)%k+100]=dp[i][j] (dp[i][j]==1) , dp[i+1][(j-data[i]-100)%k+100]=dp[i][j] (dp[i][j]==1)。
AC代码:
1 #include <cstdio> 2 int N,K; 3 int data[100010]; 4 int dp[10010][210]; 5 int main(){ 6 scanf("%d%d",&N,&K); for(int i=1;i<=N;++i) scanf("%d",&data[i]); 7 data[1]%=K; dp[2][100+data[1]]=dp[2][100-data[1]]=1; 8 for(int i=2;i<=N;++i){ 9 data[i]%=K; 10 for(int j=0;j<=K+100;++j) 11 if(dp[i][j]){ 12 dp[i+1][(j+data[i]-100)%K+100]=1; 13 dp[i+1][(j-data[i]-100)%K+100]|=dp[i][j]; 14 } 15 } 16 if(dp[N+1][100]) printf("YES"); 17 else printf("NO"); 18 return 0; 19 }
【OpenJudge3531】【背包DP】【膜法交配律】判断整除