2021牛客暑期多校训练营1 F - Find 3-friendly Integers -（ 数位dp ）

题目链接：点击进入

题目

题意

如果一个自然数存在一个子串是 3 的倍数那它就是 3-friendly 的。多组数据，求 L~R 中 3-friendly 的数的个数。

思路

对于每一个位置pos，维护一个state状态，记录这个位置之前以pos-1为右边界的连续和%3的所有结果（0、1、2）（我用二进制记录状态，0代表这位没出过，1代表出现过，第一位对应%3=0，第二位对应%3=1，第三位对应%3=2）

代码

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<stack>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
#define inf 0x3f3f3f3f
#define int long long
//#define double long double
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double pai=acos(-1.0);
const int maxn=1e6+10;
const int mod=998244353;
const double eps=1e-9;
const int N=5e3+10;
const int d1[4][2]={1,0,0,-1,0,1,-1,0};
const int d2[4][2]={1,0,0,1,0,-1,-1,0};
int a[20];
int dp[20][8];
int dfs(int pos,int state,bool lead,bool limit)
{
	if(pos==-1) return state&1;
	if(!limit&&!lead&&dp[pos][state]!=-1) return dp[pos][state];
	int up=limit?a[pos]:9;
	int res=0;
	for(int i=0;i<=up;i++)
	{
		int tmp=0,sum=0;
		if(state&1) tmp|=1;//出现过%3==0的情况 
		else
		{
			if(!(lead&&i==0))// 不是 有前导零且i==0 的情况 
			{
				tmp|=(1<<(i%3));
				if(state&(1<<1)) tmp|=(1<<((1*10+i)%3));//之前初出现过%3==1的情况 
				if(state&(1<<2)) tmp|=(1<<((2*10+i)%3));//之前初出现过%3==2的情况 
			}
		}
		res+=dfs(pos-1,tmp,lead&&i==0,limit&&i==a[pos]);
	} 
	if(!limit&&!lead) dp[pos][state]=res;
	return res;
}
int solve(int x)
{
	int pos=0;
	while(x)
	{
		a[pos++]=x%10;
		x/=10;
	}
	return dfs(pos-1,0,1,1);
}
signed main()																						
{	
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	int t,l,r;
	scanf("%d",&t);
	while(t--)
	{
		memset(dp,-1,sizeof(dp));
		scanf("%lld%lld",&l,&r);
		printf("%lld\n",solve(r)-solve(l-1));
	}
    return 0;
}