CodeForces 204 A.Little Elephant and Interval（组合数学）

Description

给出一个区间$[L,R]$，问其中首尾数字相同的数字有多少个

Input

两个正整数$L,R(1\le L\le R\le 10^{18})$

Output

输出区间$[L,R]$中首尾相同的数字个数

Sample Input

2 47

Sample Output

12

Solution

考虑求出$1$~$n$之间首尾相同数字个数即可，假设$n$的长度为$m$

对于长度$l<m$的数字，其首尾$1$~$9$随便取一个相同即可，中间$l-2$位随便取，方案数$9+\sum\limits_{l=2}^{m-1}9\cdot 10^{l-2}=10^{m-2}+8$

对于长度$l=m$但最高位小于$n$的最高位$x$的数字，只要其最高位小于$x$，中间$m-2$个数字随便取，方案数$(x-1)\cdot 10^{m-2}$

对于长度$l=m$且最高位等于$x$的数字，如果$n$的最低位$y\ge x$，那么这个数字中间$m-2$位只要不超过$n$的中间$m-2$位的值即可，方案数为$\lfloor\frac{n}{10}\rfloor - x\cdot 10^{m-2}+1$，否则中间这$m-2$个数字不能和$n$中间$m-2$位相同，方案数减一

综上，如果$y\ge x$，答案为$10^{m-2}+8+(x-1)\cdot 10^{m-2}+\lfloor\frac{n}{10}\rfloor-x\cdot 10^{m-2}+1=\lfloor\frac{n}{10}\rfloor+9$，否则答案为$\lfloor\frac{n}{10}\rfloor+8$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
ll Solve(ll n)
{
    if(n<10)return n;
    int x=n%10,y;
    ll nn=n;
    while(nn>=10)nn/=10;
    y=nn;
    if(x>=y)return n/10+9;
    return n/10+8;
}
int main()
{
    ll L,R; 
    while(~scanf("%I64d%I64d",&L,&R))printf("%I64d\n",Solve(R)-Solve(L-1));
    return 0;
}