LightOJ - 1068 Investigation（数位dp）

是羊驼鸭

于 2021-08-13 17:00:33 发布

阅读量163

点赞数

分类专栏：数位dp

本文链接：https://blog.csdn.net/qq_39641976/article/details/119657861

版权

数位dp 专栏收录该内容

10 篇文章 1 订阅

订阅专栏

题目链接：点击这里

题目大意：
多组样例，每组样例给出一个区间 $[l, r]$ 和一个 $k$ ，求区间内有多少个数字满足其本身和数位和都能被 $k$ 整除

题目分析：
由于 $k\le 1e4$ 所以如果我们开 $dp[pos][max_k][max_k]$ 这么大的数组来让 $d p [i] [j] [k]$ 表示前 $p o s$ 位的代表的数字模 $k$ 为 $j$ ，数位和模 $k$ 为 $k$ 的数字个数，会爆空间，考虑如下优化。
由于 $l, r$ 不超过 $i n t$ 范围，所以其数位和最大不会超过 $90$ （显然其在是十进制下不会有 $10$ 位），所以在 $\ge 90$ 时直接输出 $0$ 即可。
其余情况就可以直接用数位 $d p$ 做了

具体细节见代码：

//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
#define int  ll
#define endl '\n'
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0)
using namespace std;
int read()
{
	int res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 20+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
int n,k,a[25],dp[25][90][90]; 
int dfs(int pos,int sum,int pre,bool limit)
{
	if(pos == -1) return !sum && !pre;
	if(!limit && dp[pos][sum][pre] != -1) return dp[pos][sum][pre];
	int up = limit ? a[pos] : 9,res = 0;
	for(int i = 0;i <= up;i++)
	{
		res += dfs(pos-1,(sum*10+i)%k,(pre+i)%k,limit&&i==up);
	}
	if(!limit) dp[pos][sum][pre] = res;
	return res;
}
int calc(int x)
{
	int cnt = 0;
	while(x)
	{
		a[cnt++] = x%10;
		x /= 10;
	}
	return dfs(cnt-1,0,0,1);
}
signed main() 
{
	
	int t = read();
	for(int i = 1;i <= t;i++)
	{
		memset(dp,-1,sizeof(dp));
		int l = read(),r = read();k = read();
		if(k >= 90) printf("Case %lld: 0\n",i);
		else printf("Case %lld: %lld\n",i,calc(r)-calc(l-1));
	}
	return 0;
}