P5664 Emiya 家今天的饭 题解

博客园同步

原题链接

简要题意：

对每种菜有对应的若干种烹饪方法，用二维矩阵表示。求：每种食材用不超过 $\lfloor \frac{k}{2}\rfloor$ 次，至少炒一道菜，烹饪方法互不相同的个数。

算法一

注意到，对于 $32\%$ 的数据，$n\le 10$，$m\le 3$.

这部分我们可以试试暴力。

暴力的注意点是，烹饪方法是独立的，因此对每个烹饪方法进行分别计算。

具体见代码。

时间复杂度：$O((m+1{)}^n)$.

实际得分：$32pts$.

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;

const int N=2e3+1;
const int MOD=998244353;
typedef long long ll;

inline int read(){
   char ch=getchar();int f=1;while(ch<'0' || ch>'9') {
   if(ch=='-') f=-f; ch=getchar();}
	int x=0;while(ch>='0' && ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x*f;}

int h[N],n,m; //h[i] 表示烹饪方法的选择种数
ll ans=0; int a[N][N];

inline void dfs(int dep,int sum,int x,int y) {
   
    // dep 是正在决策的一行（即烹饪方法），sum 是方案数，x 是要选的行数，y 是已选的行数
	if(x==y) {
   
		ans=(ans+sum)%MOD; return;
	} if(dep>n) return;
	for(int i=1;i<=m;i++)
		if(h[i]+1<=x/2) {
    //要求不超过一半
			h[i]++;
			dfs(dep+1,(1ll*sum*a[dep][i])%MOD,x,y+1); //乘法原理
			h[i]--; //注意减回去，不要影响回溯
		} dfs(dep+1,sum,x,y); //否则不选
}

int main(){
   
	n=read(),m=read();
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++) a[i][j]=read();
	if(n<=10) {
   
		for(int i=2;i<=n;i++) dfs(1,1,i,0); //对每个烹饪方法分别计算
		printf("%lld\n",ans); return 0;
	} //第一档部分分
	return 0;
}

算法二

注意到，$m=2$ 且 $n=40$ 是一个部分分。

这时，每种烹饪方法只有 $3$ 种：不选，选第 $1$ 个，选第 $2$ 个。（不可以都选，因为不能超过 $\lfloor \frac{m}{2}\rfloor$）.

此时这就是背包。

即把 $a_{i,j}$ 当成价值，然后有 $2$ 个物品，$40$ 种选法即可。

用 $f_{i,j,k}$ 表示前 $i$ 种烹饪方法，第一个物品选了 $j$ 种，第二个选了 $k$ 种的方案数。

则：

f i , j , k = f i − 1 , j , k + f i − 1