noi 2.6_3531判断整除（DP）

Posted 2020-08-13 konjac蒟蒻

题意：给一个正整数数列，可将其相加或相减，问是否有一个结果能被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 }