[CF991E] Bus Number 解题记录

最新推荐文章于 2024-07-15 13:32:43 发布

Li_Feiy

最新推荐文章于 2024-07-15 13:32:43 发布

阅读量729

点赞数 28

文章标签：深度优先 c++ 推荐算法算法

本文链接：https://blog.csdn.net/Li_Feiy/article/details/137926356

版权

这篇文章讨论了解决编程竞赛题目的方法，涉及计算给定序列A中数字构成序列B的排列数，同时满足B中数字在A中出现且不含前导0的规则。作者使用了递归和组合数学的思想来求解，给出了C++代码实现。

摘要由CSDN通过智能技术生成

[CF991E] Bus Number 解题记录

题意简述

给定一个序列 $A$ ，求构成序列 $B$ 的方案数。满足：

$B$ 中的数字在 $A$ 中出现过。
$A$ 中数字在 $B$ 中也出现过。
不含前导 $0$ 。

题目分析

考虑先不管第 $3$ 条限制，如果当前有 $s u m$ 个数字，每个数字 $\in [0,9]$ 出现了 $a_i$ 次，那么总排列数就是 $\frac{sum!}{\prod \limits ^9 _{i=0} a_i}$ 。
接下来考虑含前导 $0$ 的情况。
由于一共有 $s u m$ 个数字，其中 $0$ 有 $a_0$ 个，那么 $0$ 开头的排列数就占所有排列数 $p$ 的 $\frac{a_0}{sum}$ 个（对于 $s u m$ 中每个数都有以它为开头的对应排列，那么以 $0$ 开头的就有 $a_0$ 个），所以就需要从 $p$ 中减去 $\times \frac{a_0}{sum}$ 。

暴力搜索 $B$ 中每个数有多少个就行。

AC Code

#include<bits/stdc++.h>
#define arrout(a,n) rep(i,1,n)std::cout<<a[i]<<" "
#define arrin(a,n) rep(i,1,n)a[i]=read()
#define rep(i,x,n) for(int i=x;i<=n;i++)
#define dep(i,x,n) for(int i=x;i>=n;i--)
#define erg(i,x) for(int i=head[x];i;i=e[i].nex)
#define dbg(x) std::cout<<#x<<":"<<x<<" "
#define mem(a,x) memset(a,x,sizeof a)
#define all(x) x.begin(),x.end()
#define arrall(a,n) a+1,a+1+n
#define PII std::pair<int,int>
#define m_p std::make_pair
#define u_b upper_bound
#define l_b lower_bound
#define p_b push_back
#define CD const double
#define CI const int
#define int long long
#define il inline
#define ss second
#define ff first
#define itn int
int read() {
    char ch=getchar();
    int r=0,w=1;
    while(ch<'0'||ch>'9') w=ch=='-'?-1:w,ch=getchar();
    while(ch>='0'&&ch<='9') r=r*10+ch-'0',ch=getchar();
    return r*w;
}
CI N=25;
std::string s;
int ans,a[N],cnt[N],fac[N];
void dfs(int x){
    if(x>9){
        int sum=0;
        rep(i,0,9){
            sum+=a[i];
        }
        int p=fac[sum];
        rep(i,0,9){
            p/=fac[a[i]];
        }
        if(a[0]){
            p-=p*a[0]/sum;
        }
        ans+=p;
        return ;
    }
    rep(i,1,cnt[x]){
        a[x]=i;
        dfs(x+1);
    }
    if(!cnt[x]){
        dfs(x+1);
    }
}
signed main() {
    std::cin>>s;
    rep(i,0,s.size()-1) {
        cnt[s[i]-'0']++;
    }
    fac[0]=1;
    rep(i,1,20){
        fac[i]=fac[i-1]*i;
    }
    dfs(0);
    std::cout<<ans;
    return 0;
}