Codeforces 991E. Bus Number (DFS+排列组合)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

string s;
int num[10];
ll jc[20];//阶乘 
int tempNum[10];
ll ans = 0;

void dfs(int x){
    //如果0~9都填充好了数字 
    if(x == 10){
        //cnt表示所有数字的个数 
        int cnt = 0;
        for(int i = 0;i < 10; ++i){
            cnt += tempNum[i];
        }
        //排列组合公式 开始 
        ll p = jc[cnt];
        for(int i = 0;i < 10; ++i){
            p /= jc[tempNum[i]];
        }
        if(tempNum[0] >= 1)
            p -= (p*tempNum[0]/cnt);
        //排列组合公式 结束 
        ans += p;
        return ;
    }
    //对于出现过的数字，个数从1开始 
    for(int i = 1;i <= num[x]; ++i){
        tempNum[x] = i;
        dfs(x+1);
    }
    if(num[x] == 0){
        dfs(x+1);
    }
}


int main(){
    ios::sync_with_stdio(false);
    cin >> s;
    for(auto i:s) num[i-'0']++;
    //算阶乘 
    jc[0] = 1;
    for(ll i = 1;i <= 19; ++i){
        jc[i] = jc[i-1]*i;
    }
    dfs(0);
    cout << ans << endl;
    return 0;   
}