B - 风之守护-CSDN博客

本文链接：https://blog.csdn.net/qq_73051419/article/details/131237870

题意：输入n，k，分别代表数的位数，以及数的进制数，要求满足该条件的数有多少个没有连续两个0的数

思路：此题不易用数位dp，因为n和k是可以开到很大的（虽然这题很小），dp[i][j]代表当前位填j时有多少种情况满足条件，由于此题的判断条件只与0有关，所以我们规定j=0代表此位填0，j=1代表此位不填0，状态转移方程见代码

由于题目规定不能有前导零，所以第一位只能选1到9，所以初始化dp[1][0]=0,dp[1][1]=k-1

ac代码：

#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
#define pb push_back
//#define int long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;
typedef pair<int,int> pii;
const int N=30;
int n,k;
int dp[N][2];
void solve()
{
	cin>>n>>k;
	dp[1][0]=0;
	dp[1][1]=k-1;
	for(int i=2;i<=n;i++)
	{
		dp[i][0]=dp[i-1][1];
		dp[i][1]=(dp[i-1][0]+dp[i-1][1])*(k-1);
	}
	cout<<dp[n][0]+dp[n][1]<<endl;
}
signed main()
{
	Mirai;
	int T=1;
	//cin>>T;
	while(T--)
	{
		solve();
	}
}