洛谷P6754 【Palindrome-Free Numbers】-数位dp

传送门

题意

定义回文串为：对于一个数字串，若其某个长度大于1的子串是回文串，那么它是回文串。
给定$a,b$，求区间$[a,b]$中有多少个非回文串。

数据范围

$a,b\in [0,10^{18}]$

题目分析

数字串包含关系的偏序的极小元为长度为2或3的回文串。当我们从高到低枚举到某一位数字时，相应的，只需要考虑高1位和高2位的数字。
设计状态dp[pos][sta][t1][t2]：当前枚举到第pos位，之前的答案是sta,高1位是t1,高2位是t2时，满足条件的数字的个数。对于起始状态（高位没有数字），可以用-1来表示起始状态。

#include<bits/stdc++.h>
#define de(x) cout<<#x<<" = "<<x<<endl;
#define rep(i,a,b) for(int i = (a);i<=(b);++i)
#define per(i,a,b) for(int i = (a);i>=(b);--i)
#define endl '\n'
#define PI acos(-1.0)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N = 25;
const int MOD = 998244353;

int digit[N],top;
ll dp[N][2][11][11];
ll dfs(int pos,bool sta,int t1,int t2,bool limit,bool zero){
	if(!pos) return sta;
	if(!limit && ~dp[pos][sta][t1+1][t2+1]) return dp[pos][sta][t1+1][t2+1];
	int up = limit ? digit[pos] : 9;
	ll res = 0;
	rep(i,0,up){
		if(zero && !i) res += dfs(pos-1,sta,-1,-1,limit&&i==up,1);
		else{
			bool _sta = sta;
			if(~t1 && t1 == i) _sta = 1;
			if(~t2 && t2 == i) _sta = 1;
			res += dfs(pos-1,_sta,i,t1,limit&&i==up,0);
		}
	}
	if(!limit) dp[pos][sta][t1+1][t2+1] = res;
	return res;
}

ll solve(ll x){
	memset(dp,-1,sizeof(dp));
	top = 0;
	while(x){
		digit[++top] = x%10;
		x/= 10;
	}
	ll res =dfs(top,0,-1,-1,1,1);
	return res;
}

int main(){
	ll x,y;
	scanf("%lld%lld",&x,&y);
	printf("%lld\n",y-x+1-solve(y) + (x != 0 ? solve(x-1) : 0));
	return 0;
}