首页 > 代码库 > 【noi 2.6_3531】判断整除(DP)

【noi 2.6_3531】判断整除(DP)

题意:给一个正整数数列,可将其相加或相减,问是否有一个结果能被K整除。

解法:似上一题“糖果”的状态定义,f[i][j]表示是否有一个选了前 i 个数的结果模K余j。

P.S. 可以用滚动数组。

 1 #include<cstdio> 2 #include<cstdlib> 3 #include<cstring> 4 #include<iostream> 5 using namespace std; 6  7 const int N=10010,K=110; 8 int a[N],f[2][K]; 9 10 int mmax(int x,int y) {return x>y?x:y;}11 int main()12 {13     int n,k;14     scanf("%d%d",&n,&k);15     for (int i=1;i<=n;i++)16       scanf("%d",&a[i]);17     memset(f,false,sizeof(f));18     int e=0;19     f[0][0]=true;20     for (int i=1;i<=n;i++)21     {22      e=1-e;23      for (int j=0;j<k;j++)24      {25        int p=(j+k-a[i]%k)%k,q=(j+k+a[i]%k)%k;26        //或p=(j-a[i]+5000*k)%k,q同理27        f[e][j]=f[1-e][p]|f[1-e][q];28      }29     }30     if (f[e][0]) printf("YES");31     else printf("NO");32     return 0;33 }

 

【noi 2.6_3531】判断整除(DP)