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