CFdiv2-Playlist for Polycarp-(二进制枚举+组合数+记忆化搜索或者暴搜)

G1

题意：
就是给你n个音乐，每个音乐有个时间和曲调。然后一共总时间为m，现在问你有多少种不同的组合使得选择的音乐总时间为m，并且任意两个音乐之间的曲调不能相同。数据范围都很小，同时取模。

思考：

1. 由于看到n才15，很明显就会去想二进制枚举或者暴搜方面的知识。对于总时间为m，我可以二进制枚举选择哪些音乐满足=m，然后看看曲调为1的选了几个为2的选了几个为3的选了几个。
2. 但是我怎么求这些音乐的组合呢？刚开始我想用球和盒子问题，但是推了推发现并不好做，所以就换思路了。想来想去，由于需要看这个音乐的顺序，也就是相同曲调的这些音乐我可以求个全排列。也就是我现在只考虑每个位置放什么曲调的音乐，而不考虑这个曲调的音乐到底选哪一个。
3. 那么就好办了，直接包搜每个位置填什么，只要不和上一位冲突就行。复杂度3ⁿ = 1e7差不多。然后这题就过了。但是对于这种搜索，完全可以记忆化，所以我定义dp[M][M][M][M][M]，当前在第i位，前一位是曲调j，并且还剩下a个曲调1，b个曲调2，c个曲调3的方案数。那么记忆化一下就好了，但是发现并没有加快多少速度。可能数据比较小不明显。

代码：

暴搜
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
PII va[N];
int vb[N];
int res;

void dfs(int now,int a,int b,int c)
{
	if(a==0&&b==0&&c==0)
	{
		res++;
		return ;
	}
	if(a>=1&&vb[now-1]!=1)
	{
		vb[now] = 1;
		dfs(now+1,a-1,b,c);
		vb[now] = 0;
	}
	if(b>=1&&vb[now-1]!=2)
	{
		vb[now] = 2;
		dfs(now+1,a,b-1,c);
		vb[now] = 0;
	}
	if(c>=1&&vb[now-1]!=3)
	{
		vb[now] = 3;
		dfs(now+1,a,b,c-1);
		vb[now] = 0;
	}
	return ;
}

int f(int x)
{
	int sum = 1;
	for(int i=1;i<=x;i++) sum = sum*i%mod;
	return sum;
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>va[i].fi>>va[i].se;
	int ans = 0;
	for(int i=0;i<(1ll<<n);i++)
	{
		int sum = 0,t1 = 0,t2 = 0,t3 = 0;
		for(int j=0;j<n;j++)
		{
			if(i>>j&1)
			{
				sum += va[j+1].fi;
				t1 += (va[j+1].se==1);
				t2 += (va[j+1].se==2);
				t3 += (va[j+1].se==3);
			}	
		}
		if(sum!=m) continue;
		res = 0;dfs(1,t1,t2,t3);
		ans = (ans+res*f(t1)%mod*f(t2)%mod*f(t3)%mod)%mod;
	}
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}
记忆化搜索:
#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 20;

int T,n,m,k;
PII va[M];
int vb[M];
int dp[M][M][M][M][M];
//当前在第i位,前一位是j,并且还剩下a个a，b个b，c个c的方案数。
int dfs(int now,int a,int b,int c)
{
	if(a==0&&b==0&&c==0) return 1;
	if(dp[now][vb[now-1]][a][b][c]!=-1) return dp[now][vb[now-1]][a][b][c];
	int res = 0;
	if(a>=1&&vb[now-1]!=1)
	{
		vb[now] = 1;
		res = (res+dfs(now+1,a-1,b,c))%mod;
		vb[now] = 0;
	}
	if(b>=1&&vb[now-1]!=2)
	{
		vb[now] = 2;
		res = (res+dfs(now+1,a,b-1,c))%mod;
		vb[now] = 0;
	}
	if(c>=1&&vb[now-1]!=3)
	{
		vb[now] = 3;
		res = (res+dfs(now+1,a,b,c-1))%mod;
		vb[now] = 0;
	}
	dp[now][vb[now-1]][a][b][c] = res;
	return res;
}

int f(int x)
{
	int sum = 1;
	for(int i=1;i<=x;i++) sum = sum*i%mod;
	return sum;
}

signed main()
{
	IOS;
	int sum = 1;
	for(int i=1;i<=15;i++) sum = sum*3;
	cout<<sum;
	return 0;
	mem(dp,-1);
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>va[i].fi>>va[i].se;
	int ans = 0;
	for(int i=0;i<(1ll<<n);i++)
	{
		int sum = 0,t1 = 0,t2 = 0,t3 = 0;
		for(int j=0;j<n;j++)
		{
			if(i>>j&1)
			{
				sum += va[j+1].fi;
				t1 += (va[j+1].se==1);
				t2 += (va[j+1].se==2);
				t3 += (va[j+1].se==3);
			}	
		}
		if(sum!=m) continue;
		ans = (ans+dfs(1,t1,t2,t3)*f(t1)%mod*f(t2)%mod*f(t3)%mod)%mod;
	}
	ans = (ans%mod+mod)%mod;
	cout<<ans;
	return 0;
}

总结：
多多简化问题，把复杂的问题简单化多多思考。