题目描述
某人刚学习了数位DP,他在某天忽然思考如下问题:
给定n,问有多少数对 <x, y>满足:
x, y∈[1, n], x < y
x, y中出现的[0, 9]的数码种类相同
输入
一个整数n (n <= 107)
输出
输出一个数即答案
样例输入
30
样例输出
3
提示
<1, 11> <2, 22> <12, 21>
思路
数字一共有0~9
10个数字,对于给出的n
我们可以遍历一遍从1~n
,对于每一个数,把当前的数拆成10进制,然后利用二进制压位,压缩成一个状态,用一个数组标记一下当前的状态出现了几次,最后求出
C2n
C
n
2
的和就行。
二进制压位的时候,当这一位存在时,就num|=(1<<i)
代码
#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
const ll N=1e7+20;
ll vis[15];
ll mp[1025];
int main()
{
ll n,len=0,num=0;
scanf("%lld",&n);
for(ll i=1; i<=n; i++)
{
mem(vis,0);
ll x=i;
while(x)
{
vis[x%10]=1;
x/=10;
}
ll num=0;
for(ll j=0; j<=9; j++)
if(vis[j])
num|=(1<<j);
mp[num]++;
}
ll sum=0;
for(ll i=1; i<=1025; i++)
if(mp[i])
sum+=(mp[i])*(mp[i]-1)/2;
printf("%lld\n",sum);
return 0;
}