P2110 欢总喊楼记

题目描述

诗经有云：

关关雎鸠，在河之洲。窈窕淑女，君子好逑。

又是一个被风吹过的夏天……一日欢总在图书馆中自习，抬起头来，只见一翩跹女子从面前飘过，真是回眸一笑百媚生，六宫粉黛无颜色！一阵诗情涌上欢总心头，他顺手写下那诗句：

啊，你是爱，是暖，是希望，你是人间四月天！

任时光匆匆而过，欢总一直没能忘记那女子。人言单思苦，欢总偏单思。夜夜难入梦，此心淑可知。偶然一次机会，欢总得知了那女孩的信息。一日，欢总终于鼓起了勇气，他向她表白！

那晚的夜色格外美丽，欢总在楼下慷慨激昂，气氛浪漫而感人。女孩有点心动了，但是直接答应是不是有点太不矜持了呢？于是，她想难难欢总，看看他到底有多少诚意。

女孩给出了两个整数 L 和 R，她要欢总数出到底有多少个这样的 X：L <= X <= R，且 X 的最高位与最低位相等（十进制下）。比如，2、101、329873可以是这样的 X，而23、4567就不是。

孩子们，欢总下辈子的幸福生活就靠你们帮忙了！

输入输出格式

输入格式：

一行，这一行包括两个整数 L 和 R。

输出格式：

一行，这一行包括一个整数，即满足所述性质的 X 的个数。

输入输出样例

输入样例#1： 复制

2 47

输出样例#1： 复制

12

说明

【数据规模】

50% 1<=L<=R<=10^6

100% 1<=L<=R<=10^18

---

数位dp，非常简单的那种

一定要数好数据范围！！！ 记得开LL！！！！

---

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cstring>

#define LL long long

using namespace std;

int w[20],e[20],i,m,n,j,k;

LL p,q,ans,f[20][10][2][2];

int main()

{

	scanf("%lld%lld",&p,&q);

	LL t=q; k=0;

	while(t) k+=1, t/=10;

	t=q;

	for(i=k;i>=1;i--) w[i]=t%10, t/=10;

	t=p;

	for(i=k;i>=1;i--) e[i]=t%10, t/=10;

	for(i=e[1];i<=w[1];i++)

	{

		int l=0,r=0;

		if(i==e[1]) l=1;

		if(i==w[1]) r=1;

		f[1][i][l][r]=1;

		if(k==1) ans+=1;

	}

	for(int i=2;i<=k;i++)

	for(int j=0;j<=9;j++)

	for(int l=0;l<=1;l++)

	for(int r=0;r<=1;r++)

	for(int z=0;z<=9;z++)

	if(f[i-1][j][l][r])

	{

		if(l && z<e[i]) continue;

		if(r && z>w[i]) continue;

		int ll=0, rr=0, las=0;

		if(j==0) las=z;

		else las=j;

		if(l && z==e[i]) ll=1;

		if(r && z==w[i]) rr=1;

		f[i][las][ll][rr]+=f[i-1][j][l][r];

		if(i==k && z==las) ans+=(LL)f[i-1][j][l][r];

	}

	printf("%lld",ans);

}