牛客练习赛40-题解(部分)
官方题解:https://www.nowcoder.com/discuss/154293?type=101
A-小D的剧场
大概意思就是,有1~49个数,代表着不同的音符,问你能组成长度为n的串有多少种(mod 109+7)。有若干个限制,比如1 2 3,就是1,2,3不能放在一起,然后问你在这些限制下有多少种不同的长度为n的串。
3≤n≤500,0≤q≤117649,1≤a,b,c≤49
思路
看到这题,想到HDU2047,不难想到dp的方法,就是一个暴力的dp题。dp[i][j][k]代表长度为i,第i个位置为j,第i-1个位置为k的种数,状态转移方程为:dp[i][j][k]+=dp[i-1][k][l],预处理好vis[j][k][l]是否能放在一起。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
ll n,q;
ll a,b,c;
ll vis[50][50][50];
ll dp[510][50][50];
int main(){
memset(vis,0,sizeof(vis));
memset(dp,0,sizeof(dp));
scanf("%lld %lld",&n,&q);
for(ll i=0; i<q; i++){
scanf("%lld %lld %lld",&a,&b,&c);
vis[a][b][c]=1;
vis[a][c][b]=1;
vis[b][a][c]=1;
vis[b][c][a]=1;
vis[c][a][b]=1;
vis[c][b][a]=1;
}
for(int i=1; i<50; i++)