OpenJudge_P3531 判斷整除(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

#include<bits/stdc++.h>
using namespace std;
const int N=105+7;
const int inf=0x3f3f3f3f;
const int mod=1000000007;
int n,m,k,flag,b=0;
int f[2][N];
int mods(int x)
{
    return x>=0?x%k:x%k+k;
}
int main()
{
//    freopen("C:\\Users\\lenovo\\Desktop\\1.in","r",stdin);
    int t,tt=0,x;
    scanf("%d%d",&n,&k); flag=1;

    while(n--)
    {
        scanf("%d",&x); x=mods(x);
        memset(f[b],0,sizeof f[b]);
        
        if(flag) flag=0,f[b][mods(x)]=f[b][mods((-1)*x)]=1;
        
        for(int i=0;i<k;i++)
            if(f[!b][i])
            f[b][mods(i+x)]=f[b][mods(i-x)]=1;
        b=!b;
    }
    printf("%s\n",f[!b][0]?"YES":"NO");
    return 0;
}