数位dp

当题目中限制数字某一些位之间的关系时，一般用到数位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;
}