poj 1745 Divisibility dp

点击打开题目链接

题意：给出n个数字，让你在这n个数字之间加上加号和减号，问最后能不能得到一个数字，使它能被k整除。

思路：dp，把每个数字处理后可能得到的结果保存下来，如果能得到，dp值就是1，否则就是0.需要注意的是，每个得到的值必须是由前面的所有数字进行加减得到的，所以每次状态转移的时候只能把新得到的状态存下来，而之前的状态都抛弃，因为之前得到的状态都是只由前i个数字得到的，不和第i+1个数字由关系，所以可以用滚动数组来存，减少空间花费。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
using namespace std;

bool dp[210][2];
int a[10010];

int main()
{
    int n,k;
    while(cin>>n>>k){
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        dp[a[0]%k+k][0]=1;
        for(int i=1;i<n;i++){
            for(int j=0;j<210;j++)
                dp[j][i%2]=0;
            for(int j=1;j<=2*k-1;j++) if(dp[j][1-i%2]){
                j-=k;
                int x=(j+a[i])%k;
                dp[x+k][i%2]=1;
                x=(j-a[i])%k;
                dp[x+k][i%2]=1;
                j+=k;
            }
        }
    if(dp[k][1-n%2]) cout<<"Divisible"<<endl;
    else cout<<"Not divisible"<<endl;
    }
    return 0;
}