算法笔记_数位DP

windy 数

题意：

不含前导零且相邻两个数字之差至少为 2 的正整数被称为 windy 数。

询问在区间$[a,b]$中有多少 windy数

解析：

需要有一维记录当前状态的前一位的数。

注意前导 $0$，如果当前状态的前一位为前导 $0$ 时，当前位可以从 $0$ 开始枚举。当有前导零且当前位也为 $0$ 时，可以将当前值赋为 $-2$ 传到给下一位的 $dfs$(具体可以看代码)

代码：

#include<iostream>
#include<cstring>
#include<cmath> 
#include<vector>
using namespace std;
typedef long long ll;
int dp[20][15];
vector<int> digit;
ll dfs(int pos, int pre, int lead, int limit){
	if(pos == -1) return 1;
	if(!limit&&!lead&&dp[pos][pre]!=-1) return dp[pos][pre];
	ll tmp = 0;
	int op;
	int up = limit?digit[pos]:9;
	for(int i = 0; i <= up; i++){
		if(abs(i-pre)<2) continue;
		op = i;
		if(!i&&lead) op = -2;
		tmp += dfs(pos-1, op, op==-2,limit&&i==up);
	}
	if(!limit&&!lead) dp[pos][pre] = tmp;
	return tmp;
}
ll solve(ll num){
	memset(dp, -1, sizeof(dp));
	digit.clear();
	ll res;
	while(num){
		digit.push_back(num%10);
		num /= 10;
	}
	return dfs(digit.size()-1, -2, 1, 1);
}
int main(){
	int a, b;
	cin >> a >> b;
	cout << solve(b) - solve(a-1);
	return 0;
}

数字计数

题意：

统计 $[l,r]$ 中每个数码(digit)各出现了多少次。

解析：

要记录的信息有：当前位置，前导$0$，上界，数码 $d$ 出现的次数

代码：

#include<iostream>
#include<cstring>
using namespace std;
const int N=15;
long long le,ri,dp[N][N],num[N];
int d,cnt;
long long dfs(int pos,long long sum,int d,bool limit,bool lead){
	if(pos==0) return sum;
	if(!limit&&!lead&&dp[pos][sum]!=-1) return dp[pos][sum];
	int up=limit? num[pos]:9;
	long long tmp=0;
	for(int i=0;i<=up;i++){
		tmp+=dfs(pos-1,sum+((i==d)&&(!lead||i)),d,limit&&i==num[pos],i==0&&lead);
	}
	dp[pos][sum]=tmp;
	return tmp;
}
long long solve(long long x,int d){
	cnt=0;
	while(x){
		num[++cnt]=x%10;
		x/=10;
	}
	return dfs(cnt,0,d,1,1);
}
int main(){
	cin>>le>>ri;
	for(int i=0;i<=9;i++){
		memset(dp,-1,sizeof(dp));
		cout<<solve(ri,i)-solve(le-1,i);
		if(i!=9) cout<<" ";
	}
	return 0;
}

Sam数

题意：

相邻两位数字之差不超过2的数为Sam 数。询问$n$位的Sam 数的个数。

解析：

令$f(i,j)$为末位为 $j$ 的 $i$ 位Sam 数的个数$$f(i,j)=\sum _{x=j-2}^{j+2}f(i-1,x)$$
此时，时间复杂度为$O(n)$，会T
需要矩阵加速递推。构造转移矩阵和初始矩阵，然后矩阵快速幂

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 15;
const int mod = 1e9+7;
struct matrix{
	ll a[maxn][maxn];
	matrix(){
		memset(a, 0, sizeof(a));
	}
	void matrixI(){
		for(int i = 0; i < 10; i++)
			a[i][i] = 1;
	}
	matrix operator * (const matrix b){
		matrix c;
		for(int k = 0; k < 10; k++)
			for(int i = 0; i < 10; i++)
				for(int j = 0; j < 10; j++)
					c.a[i][j] = (c.a[i][j]+a[i][k]*b.a[k][j]%mod)%mod;
		return c;
	}
};
matrix qpow(matrix a, ll b){
	matrix res;
	res.matrixI();
	while(b){
		if(b&1)
			res = res*a;
		b = b >> 1;
		a = a*a;
	}
	return res;
}
matrix a, b;
ll n;
ll ans;
int main(){
	cin >> n;
	if(n == 1){
		cout << 10 << endl;
		return 0;
	}
	for(int i = 1; i < 10; i++)
		a.a[0][i] = 1;
	for(int i = 0; i < 10; i++){
		for(int j = i-2; j <= i+2; j++){
			if(j < 0 || j > 9)
				continue;
			b.a[j][i] = 1;
		}
	}
	a = a*qpow(b, n-1);
	for(int i = 0; i < 10; i++)
		ans = (ans+a.a[0][i])%mod;
	cout << ans << endl;
	return 0;
}

Beautiful numbers

题意：

对于一个数 $a$ ，如果 $a$ 能被 $a$ 的每一位上非零数 $b$ 整除即 $b|a$，则 $a$ 是美丽数。

询问$[l,r]$中美丽数的个数

解析：

如果每一位都能整除 $a$，那么这些数的最小公倍数也能整除 $a$，1-9 的最小公倍数为2520，有一维记录当前状态的最小公倍数。

还需要一维来记录当前状态的 $a$，关注的是 $a$ 是否能被当前的最小公倍数 $lcm$ 整除，所以只需要记录 $a\%2520$ 的值。

此时状态为$dp[20][2520][2520]$，还是开不了。此时发现最小公倍数那一维空间浪费很多，因为最小公倍数不会取遍1-2520的所有值，最小公倍数一定是2520的因数，而2520的因数只有48个，所以空间开50就够了。

综上，$dp[20][2520][50]$

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 2520;
ll dp[20][mod+10][50];
int factor[mod+10];
ll gcd(ll a, ll b){
	if(b == 0)
		return a;
	else
		return gcd(b, a%b);
}
ll lcm(ll a, ll b){
	return a/gcd(a, b)*b;
}
vector<int> L, R;
ll l, r;
ll dfs(vector<int> v, int pos, int x, int lc, int limit){
	if(pos == -1)
		return x % lc == 0;
	if(dp[pos][x][factor[lc]] != -1 && !limit)
		return dp[pos][x][factor[lc]];
	ll tmp = 0;
	int up = limit ? v[pos] : 9;
	for(int i = 0; i <= up; i++){
		int nx = (x*10+i)%mod;
		int nlc = i == 0 ? lc : lcm(lc, i);
		tmp += dfs(v, pos-1, nx, nlc, limit && i==up);
	}
	if(!limit)
		dp[pos][x][factor[lc]] = tmp;
	return tmp;
}
void solve(){
	cin >> l >> r;
	l--;
	L.clear(); R.clear();
	while(l){
		L.push_back(l%10);
		l /= 10;
	}
	while(r){
		R.push_back(r%10);
		r /= 10;
	}
	cout << dfs(R, R.size()-1, 0, 1, 1) - dfs(L, L.size()-1, 0, 1, 1) << endl;
	return;
}
int main(){
	ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    memset(dp, -1, sizeof(dp));
    int tot = 0;
    for(int i = 1; i <= mod; i++){
    	if(mod%i == 0)
    		factor[i] = ++tot;
	}
	int T;
	cin >> T;
	while(T--)
		solve();
	return 0;
}