信息学奥赛一本通：1195：判断整除

1195：判断整除

【题目描述】

一个给定的正整数序列，在每个数之前都插入+号或−号后计算它们的和。比如序列：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;
int a[10005];
int main()
{
    int m,n,i,k;
    int sum=0;
    scanf("%d %d",&m,&n);
    for(i=0;i<m;i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];//先求出总和 
    }
    if(sum%n==0)//如果能整除，输出 
    {
    printf("YES");
        return 0;
        }
    k=sum%n;
    for(i=0;i<m;i++)//然后依次减一个 
    {
        if(2*a[i]%n==k)//判断 a[i]为负值时 
        {
           printf("YES");        
           return 0;
           }
        sum=(sum%n+2*n-2*a[i]%n)%n;
        if(sum==0)
        {
            printf("YES");
            return 0;
        }
        
    }
    printf("NO");
    return 0;
    
}

正解：从0开始加一个数和减一个数   求余

//f[i][j]:i是几个数，j是a[i]%k的余数 
//f[i][j]:=前i个数模k的值为j
//从0开始，取第一个数时，取加或是减两种情况；只在上次%k后有余的(j)列加或减a[i]，再取余，相应位上为1
//这样到n个数时，只要看n行0列上（能整除才能余为0）是不是1 
#include<bits/stdc++.h>
using namespace std;
int n,k,a[10010],f[10010][105];
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
    cin>>a[i];
    a[i]%=k;
    }
    f[0][0]=1;//0能除任何数 
    for(int i=1;i<=n;i++)//每个数 
        for(int j=0;j<k;j++)//余数 
         if(f[i-1][j])//如果上个数时，j位上是1（即余j时，要加或是减a[i]） 
        {
            f[i][(j+a[i])%k]=1;
            f[i][(j-a[i]+k)%k]=1;//加k，防负值 
        }
    if(f[n][0])    cout<<"YES";
    else           cout<<"NO";
    /*for(int i=0;i<=n;i++){
        for(int j=0;j<k;j++)
        cout<<f[i][j]<<" ";
        cout<<endl;
    }*/
    return 0;
}