洛谷 P1356 数列的整除性

解题思路

首先不难想到，如果选择用累加的方式求和然后判断是否整除一定会超时，因此不妨将“和能被常数k整除”转化为“结果除以常数k的余数为0”, 这样依次判断前i个数添加符号后的数列和的余数即可。考虑到添加数列中可能出现负数，在读入数据的时候应当先将取绝对值（用algorithm头文件里的abs函数）。

接下来用动态规划的思想解决问题。具体来说，对于每个数，我们可以选择将它加或者减，从而得到新的余数。因此，我们可以采用二维状态数组 left，其中 left[i][j] 表示前 i 个数是否可以得到余数 j。初始化时，第一个数的余数可以是 datum[1] 或者 k - datum[1]，因此有 left[1][(datum[1])%k] = true 和 left[1][k-(datum[1])%k] = true。

接下来，对于每个数，我们可以枚举前面的状态，并根据当前数的值更新状态数组。具体来说，如果前面某个状态 left[i][j] 是可达的，那么我们可以通过加或者减当前数得到新的余数 temp，并将 left[i+1][temp] 和 left[i+1][k-temp] 设置为可达。

最后，如果 left[n%2][0] 为真，那么说明可以得到余数为0，即整个序列可以被 k 整除；否则，不能。

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

int datum[10005];

int main()
{
    int num, k, n;
    cin >> num;
    bool left[2][101];          //运用滚动数组

    for(int p = 0; p < num; p++)
    {
        cin >> n >> k;
        memset(left, false, sizeof(left));
        for(int i = 1; i <= n; i++)
        {
            int temp;
            cin >> temp;
            datum[i] = abs(temp);                   //取绝对值
        }

        left[1][(datum[1])%k] = true;
        left[1][k-(datum[1])%k] = true;

        for(int i = 1; i < n; i++)
        {
            memset(left[(i+1)%2], false, sizeof(left[(i+1)%2])); //初始化滚动数组
            for(int j = 0; j < k; j++)              
            {
                if(left[i%2][j] == true)         //采用的不是背包问题的倒推，而是正推；
                {                                //求的不是背包问题的最值，而是验证是否存在；
                    int temp = (datum[i+1]+j)%k;
                    left[(i+1)%2][temp] = true;
                    left[(i+1)%2][k-temp] = true;
                }
            }
        }
        if(left[n%2][0] == false)
            cout << "Not divisible" << endl;
        else
            cout << "Divisible" << endl;
    }
    return 0;
}