链接: 我爱吃烧烤
题意:
总共 n 家烧烤店,有 m 家特殊烧烤店,从 i 到 j 号烧烤点有 mp [ i ] [ j ] 种方案,消耗一点体力值,问消耗 q 点体力值且至少经过 k 家特殊烧烤店的方案数。
思路:
- dp 记录 当前花费体力值,当前所在烧烤店编号,还有当前走过的特殊烧烤店,当前走过的烧烤店状压一下就好了,最多才 10 家。所以空间复杂度 50 *50 *1024 ,还是可以接受的,然后在转移的时候 ,还要枚举当前烧烤店是由 上一次的哪一家烧烤店转移过来的 ,所以 时间复杂度 50 * 50 *50 *1024 (1e8)一开始还以为跑不了。。。
- 最后记录一下 经过 大于等于 k 家烧烤店的答案。
代码:
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<cmath>
#include<string>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=4e+7;
const int mod=20190802;
int n,m,k,q;
ll dp[60][60][1200];
int a[100],mp[60][60],vis[100];
int f(int x) {
int ans=0;
for(int i=0; i<m; i++) {
if((x>>i)&1)
ans++;
}
return ans;
}
int main () {
memset(vis,-1,sizeof(vis));
scanf("%d%d%d%d",&n,&m,&k,&q);
for(int i=0; i<m; i++) {
scanf("%d",&a[i]);
vis[a[i]]=i;
}
for(int i=1; i<=n; i++) {
for(int j=1; j<=n; j++) {
scanf("%d",&mp[i][j]);
}
}
dp[0][1][0]=1;
for(int i=1; i <= q; i ++) {
for(int j=1; j <=n ; j++) {
for(int sta = 0; sta < (1 << m); sta ++) {
for(int h=1; h <= n; h++) {
if( vis[j] != -1 && ((sta >> vis[j]) & 1)==0 ) {
dp[i][j][sta | (1<<vis[j])]+=(dp[i-1][h][sta]*mp[h][j])%mod;
} else {
dp[i][j][sta] += (dp[i-1][h][sta] * mp[h][j])%mod;
}
}
}
}
}
ll ans=0;
for(int i=1; i<=n; i++) {
for(int sta=0; sta<(1<<m); sta++) {
if(f(sta)>=k){
ans=(ans+dp[q][i][sta])%mod;
}
}
}
printf ("%lld\n",ans);
}