洛谷
这道题 询问的是 子序列的和是否能被m整除,可以转化成 子序列的和 mod m 为零。
如果出现两个前缀和的mod 相同。说明他们中间的这一段数字 肯定是 m的倍数。
因为长度为n的数组,有n个非空前缀和,而mod m的余数,一共有m个,从0 到m-1.
所以将这m个数分配到n个前缀和中,所以只要n>m就会出现 至少两个前缀和的mod相等。所以一定有符合题意的子序列。当n==m时,假如n中有m种状态,那么必有余数为0的,成立。如果小于m中状态,那么就会有至少两个前缀和的mod值相等,符合条件。(我们这里讨论前缀和,其实将 是更苛刻的条件,因为这样相当于 挑选的子序列必须是连续的)
所以n>=m是答案是yes。
之后n<m的情况,子序列的问题,经典的dp问题了。
dp[i][j]是bool 类型,代表的是 考虑前i个物品,是否能产生余数为j
递推:dp[i] 是由dp[i-1]推过来的。
对于物品i来说,dp[i][ve[i]] 为true,相当于只选ve[i]这一个元素。
如果dp[i-1][j]为真,那么dp[i][j]也为真,相当于不选ve[i].
dp[i][(j+ve[i])%m]也为真,相当于选了ve[i]
初始化:由递推公式 可以知道 我们要处理出来dp[0]的值,一开始初始化为0,这一步直接合并到for循环里面去了。
最后看考虑n个元素,也就是考虑所有的元素,余数为0的时候,若为true则存在答案,否则无答案。
#include<bits/stdc++.h>
int main()
{
int n,m;std::cin>>n>>m;
std::vector<int>ve(n);
for (int i=0;i<n;i++)
{std::cin>>ve[i];
ve[i]=ve[i]%m;
}
if (n>=m){
std::cout<<"YES"<<"\n";
return 0;
}
//dp[i][j]代表 考虑前i个元素,他们能产生余数为j
std::vector<std::vector<bool>>dp(n,std::vector<bool >(m,0));
for (int i=0;i<n;i++)
{
dp[i][ve[i]]=true;
for (int j=0;j<m;j++)
{
if (i==0) continue;
if (dp[i-1][j]==true)
{dp[i][j]=true;
dp[i][(j+ve[i])%m]=true;
}
}
}
if (dp[n-1][0])
std::cout<<"YES\n";
else std::cout<<"NO\n";
return 0;
}