郑轻邀请赛 2266: number(集合)

给定n，问有多少数对 x, y满足：

x, y∈[1, n], x < y

       x, y中出现的[0, 9]的数码种类相同

输入

一个整数n (n <= 107)
输出

输出一个数即答案
样例输入

30

样例输出

3

提示

<1, 11> <2, 22> <12, 21>

下面代码感觉比较优了,但还是会超时,有新思路会再更新

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define pb push_back
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep1(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
const int N=1e7+100;
int vis1[N];
int arr[10];
int main()
{
    ll ans=0;
    int n;
    cin>>n;
    map<set<int>,ll>mp;
    for(int i=1;i<=n;i++)
    {
        set<int>st;
        int t=i;
        while(t)
        {
            int a=t%10;
            if(!st.count(a))
            st.insert(a);
            t/=10;
        }
        ll sum1=0;
        for(set<int>::iterator it=st.begin();it!=st.end();it++)
            sum1=sum1*10+*it;
        ans+=vis1[sum1];
        vis1[sum1]++;
    }
    cout<<ans<<endl;
    return 0;
}