【CF1267K】Key Storage——数论 排列组合 c++

#include<bits/stdc++.h>
#define int long long
using namespace std;
int b[100];
int bl;
int c[100][100];//j选i个 
void init()//预处理C 
{
    for(int j=0;j<=20;j++) c[0][j]=1;
    for(int i=1;i<=20;i++)
    {
        for(int j=i;j<=20;j++)
        {
            c[i][j]=c[i-1][j-1]+c[i][j-1];
        }
    }
}
void fen(int x)//x对应的序列 
{
    int k=2;
    while(x)
    {
        b[x%k]++;
        x/=k;
        k++;
        
        bl++;//bl是序列的总元素数量 
    }
    return;
}
int pin()
{
    int ans=1;
    int s=0;
    for(int i=20;i>=0;i--)
    {
        if(!b[i]) continue;
        ans*=c[b[i]][min(bl-i+1,bl)-s];//排列组合 
        s+=b[i];
    }
    return ans;
}
signed main()
{
    init();
    int T;
    cin>>T;
    while(T--)
    {
        int n;
        cin>>n;
        memset(b,0,sizeof(b));
        bl=0;
        fen(n);
        int x=pin();
        int y=0;
        if(b[0])
        {
            b[0]--;
            bl--;
            y=pin();//最后一位一定是零的情况 
        }
        cout<<x-y-1<<endl;
    }
    return 0;
}