这道题我首先的思路是dfs,确实可以做,但是N上了30之后就运行超时了。说明dfs还是蛮容易超时的,适合小数据。
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
long long allsum=0;
int cnt=0;
int N;
void dfs(int i,long long sum)
{
if(sum==allsum/2)
{
cnt++;
return;
}
if(sum>allsum/2)
return;
for(int j=i-1;j>=1;j--)
{
sum+=j;
dfs(j,sum);
sum-=j;
}
}
int main()
{
cin>>N;
for(int i=1;i<=N;i++)
allsum+=i;
if(allsum%2!=0)
{
cout<<0;
return 0;
}
dfs(N,N);
cout<<cnt;
return 0;
}
后来去网上看了看其他人的思路,发现妈耶,可以用dp做。
因为这道题其实也可以类似成“装东西”的题,所以很像01背包的感觉。因此,重点是要找到dp数组下标和数值含义以及状态转移方程。
神奇的思路在于,当dp[j]下标j表示当前子集的和为j,数值代表有多少个和为j的子集,那么dp[j]=dp[j]+dp[j-i] ,这个i就是当前我应该放入的数。
分析:每个数字只用到一次,每种情况的存在数可以由之前的存在的数来递推得到,01背包的变形。
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
int allsum=0;
int cnt=0;
int N;
long long dp[10000];
int main()
{
memset(dp,0,sizeof(dp));
cin>>N;
for(int i=1;i<=N;i++)
allsum+=i;
if(allsum%2!=0)
{
cout<<0;
return 0;
}
dp[0]=1; //!!!!空集也是集合!,重点是下面的循环里当j=i时,dp[i]+=dp[0]的含义就是一个子集只要i一个就行,所以算一个方案数 加上。
for(int i=1;i<=N;i++) //向背包中(一定)放入i
{
for(int j=allsum/2; j>=i; j--)
{
dp[j]+=dp[j-i];
}
}
cout<<dp[allsum/2]/2;
return 0;
}