和2989:糖果 很相像
总之这个递推就是状态转换,由已知推未知 还是
由已知和未知推已知(先将未知与已知比较)
两者都要初始化一些状态,因为由已知推未知嘛
后者由于前提中有未知,所以除了必要的初始化,其余状态都要初始化为不存在
前者不需要初始化为不存在状态,只要稍加判断已知状态是否存在,
就可以放心推出未知状态
#include <bits/stdc++.h>
using namespace std;
int a[10005];
bool dp[10005][105];//dp[i][j]表示前i个数的和 对k取余 是否等于 j
//最终求dp[n][0]
int main(){
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];
}
memset(dp,0,sizeof(dp));
//dp[1][abs(a[1])%k]=1;
//dp[1][abs(-a[1])%k]=1;
dp[1][(k+(a[1])%k)%k]=1;
dp[1][(k+(-a[1])%k)%k]=1;
for(int i=2;i<=n;i++){
for(int j=0;j<k;j++){//主角dp[i][j],可以转化成哪些状态,可以由哪些状态转化而来
if(dp[i-1][j]){//如果存在dp[i-1][j]这个状态
//dp[i][abs(j+a[i])%k]=1;
//dp[i][abs(j-a[i])%k]=1;
dp[i][(k+(j+a[i])%k)%k]=1;
dp[i][(k+(j-a[i])%k)%k]=1;
}
}
}
if(dp[n][0])cout<<"YES"<<endl;
else cout<<"NO"<<endl;
return 0;
}
要看几个数对k求余是否等于j
只要将这些数分别除以k得到的余数相加的和
对k求余是否等于j
dp[1][(k+(a[1])%k)%k]=1;
dp[1][(k+(-a[1])%k)%k]=1;
-2%5=-2 2%5=2
cout<<2%5<<endl;//2
cout<<abs(-2)%5<<endl;//-2
cout<<(-2%5+5)%5<<endl;//3
按照以上求余数之和 再对除数取余的想法还是采用第二种方法较好
对每个数求余数,为了防止得到的余数是负数,每得到余数加上k再对k求余
最后再累加这所有求得的余数
这就是传说中的 【膜法交配律】
膜法交配律,其实就是模法分配律,即
(a+b+c)%k=(a%k+b%k+c%k)%k
所以预处理输入数据全部模k。
上链接