当题目中限制数字某一些位之间的关系时,一般用到数位dp
仅限制了一个数字
windy定义了一种windy数。不含前导零且相邻两个数字之差至少为2的正整数被称为windy数。 windy想知道,
在A和B之间,包括A和B,总共有多少个windy数?
在本题中,我们只需要在范围内找一个值,满足条件即可。这种情况下我们一般高位枚举到低位,dp[pos][sta]就是枚举到pos位,状态为sta(根据题目的要求状态不同),后面已经没有flag和lead的限制(即后面随便填且前导0的情况已经消除)的答案值。
#include <iostream>
#include <cstring>
#include <vector>
#include <cmath>
using namespace std;
typedef long long ll;
vector<int> num;
ll dp[30][20]; //dp[i][j]记录处理到第i位,前一位是j不受约束情况下符合条件的值
ll dfs(ll pos,ll pre,int flag,int lead)
//处理到pos位置,前面一个数字为pre。
//flag=1表示前面的数等于最大值,当前能填的数受限(最大填到上限,否则会大于上限)
//flag=0表示前面有一个数小于上限值,当前数随便填都不会超上限
//lead表示pre是否是前导0
{
ll max_num; //当前位可用的最大值
if( pos == -1 ) return 1; //处理完成
if( !flag && !lead && dp[pos][pre] != -1 ) return dp[pos][pre]; //本次不受限,当前有值,且dp[pos][pre]已经算出来了
if( flag ) max_num = num[pos];
else max_num = 9;
ll res = 0;
for (int i = 0; i <= max_num; i++)
{
if( abs(i-pre) >= 2 || lead ) //只要是前导0就不受定义的影响了
{
res += dfs(pos-1,i,flag && (i==max_num),lead && (i == 0));
//位置往下减,当前填i,如果当前被限制且这一位等于最大值,那么下次才受限
//必须是前面是前导0且当前这位还是0,lead才能等于1
}
}
if( !flag && !lead ) dp[pos][pre] = res;
//如果没有限制且前导0已经消除
return res;
}
ll solve(ll x)
{
num.clear();
while( x ) //求出每一位
{
num.push_back(x%10);
x /= 10;
}
return dfs(num.size()-1,-2,1,1); //从处理最高位开始
}
int main()
{
ll a,b;
cin >> a >> b;
memset(dp,-1,sizeof(dp));
cout << solve(b) - solve(a-1) << '\n';
return 0;
}
限制了两个数字
T次询问,每次给出A,B,K,W,求满足下面条件的(x,y)对数:
1.x,y是整数
2.x ∈ [ 0 , A ] , y ∈ [ 0 , B ]
3.∣ x − y ∣ ≤ K&&∣x−y∣≤K
4.x^y ≤ W
A , B , K , W ≤ 1e9 , T ≤ 2000
在本题中,我们需要找到范围内的两个值,来满足题目的条件。这时候的数位dp就不能按往常这样定义了。一般就是从低位到高位枚举,dp[pos][x1][x2][sta]表示枚举到pos为,x是否在值域内,y是否在值域内,状态为sta的方案数。
#include <iostream>
#include <cstring>
using namespace std;
typedef long long ll;
int a,b,k,w;
ll dp[32][2][2][2][2][2][2][2];
int ax[32],bx[32],kx[32],wx[32];
int up = 31;
//本题由于需要处理异或,所以是二进制枚举
ll dfs(int pos,int x1,int x2,int y1,int y2,int z1,int z2,int lx)
//枚举到pos位,x是否<=a,y是否<=b,x+k是否>=y,y+k是否>=x,x+k是否有进位,y+k是否有进位,x^y是否<=w
{
if( pos == up )
{
if( x1 && x2 && (y1||z1) && (y2||z2) && lx ) return 1;
return 0;
}
if( dp[pos][x1][x2][y1][y2][z1][z2][lx] != -1 ) return dp[pos][x1][x2][y1][y2][z1][z2][lx];
ll res = 0;
for (int i = 0; i <= 1; i++)
{
for (int j = 0; j <= 1; j++)
{
bool nx1 = (x1&&(i==ax[pos])) || (i < ax[pos]);
bool nx2 = (x2&&(j==bx[pos])) || (j < bx[pos]);
bool t1 = (i+z1+kx[pos])%2,t2 = (j+z2+kx[pos])%2;
bool nz1 = (i+z1+kx[pos])>>1,nz2 = (j+z2+kx[pos])>>1;
bool ny1 = (y1&&(t1==j)) || (t1>j);
bool ny2 = (y2&&(t2==i)) || (t2>i);
bool nlx = (lx&&(i^j)==wx[pos]) || ((i^j)<wx[pos]);
res += dfs(pos+1,nx1,nx2,ny1,ny2,nz1,nz2,nlx);
}
}
return dp[pos][x1][x2][y1][y2][z1][z2][lx] = res;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t;
cin >> t;
while( t-- )
{
cin >> a >> b >> k >> w;
memset(dp,-1,sizeof(dp));
memset(ax,0,sizeof(ax));
memset(bx,0,sizeof(bx));
memset(kx,0,sizeof(kx));
memset(wx,0,sizeof(wx));
for (int i = 0; i <= 30; i++) //二进制表示
{
ax[i] = (a>>i)&1;
bx[i] = (b>>i)&1;
kx[i] = (k>>i)&1;
wx[i] = (w>>i)&1;
}
cout << dfs(0,1,1,1,1,0,0,1) << '\n';
}
return 0;
}