CodeForces 520 E.Pluses everywhere（组合数学+dp）

最新推荐文章于 2024-09-26 11:01:02 发布

v5zsq

最新推荐文章于 2024-09-26 11:01:02 发布

阅读量501

点赞数

本文链接：https://blog.csdn.net/V5ZSQ/article/details/79021304

版权

Code Forces 同时被 3 个专栏收录

606 篇文章

订阅专栏

202 篇文章

订阅专栏

组合数学

190 篇文章

订阅专栏

探讨了如何计算一个数字串在插入特定数量加号后形成的所有可能表达式的值之和，并通过组合数学和动态规划的方法给出了高效的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description

给出一个长度为 $n$ 的数字串，现在要往其中加 $k$ 个 $+$ 将其变成表达式，要求 $+$ 把串分成 $k$ 个数字，可以有前导零，每个数字长度至少为 $1$ ，对于一个表达式可以计算其值，问所有方案对应的值的和

Input

第一行输入两个整数 $n,k$ 表示串的长度和要加入的 $+$ 数量，之后输入一个长度为 $n$ 的数字串 $(1\le k<n\le 10^5)$

Output

输出所有方案对应的值的和，结果模 $10^9+7$

Sample Input

3 1
108

Sample Output

Solution

设 $n$ 个数字从高位到低位为 $a_0,...,a_{n-1}$ ，考虑 $a_i$ 对答案的影响时不用考虑 $a_i$ 前面加号的位置，只需要考虑后面第一个加号的位置，设后面第一个加号的位置在 $a_j$ 和 $a_{j+1}$ 之间，则其对答案的贡献为 $10^{j-i}$ ，而还有 $n+i-j-2$ 个位置要放 $k-1$ 个加号，方案数 $C_{n+i-j-2}^{k-1}$ ，还有一种情况是 $a_i$ 后面没有加号，其对答案的贡献是 $10^{n-i-1}$ ，那么前面 $i$ 个位置要放 $k$ 个加号，方案数 $C_i^k$ ，故 $ans=\sum\limits_{i=0}^{n-1}a_i\sum\limits_{j=i}^{n-2}10^{j-i}\cdot C_{n+i-j-2}^{k-1}+\sum\limits_{i=0}^{n-1}a_i\cdot 10^{n-i-1}\cdot C_i^k$ ，预处理 $10^x$ 和组合数，后一部分可以 $O(n)$ 得到，前一部分要化简一下，考虑枚举 $j-i$ 则前一部分可以变成 $\sum\limits_{i=0}^{n-2}10^i\cdot C_{n-i-2}^{k-1}\sum\limits_{j=0}^{n-2-i}a_j$ ，令 $s_i=\sum\limits_{j=0}^{i}a_j$ ，则可以 $O(n)$ 得到第一部分答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
int fact[maxn],inv[maxn],f[maxn];
void init(int n=1e5)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
    f[0]=1;
    for(int i=1;i<=n;i++)f[i]=(ll)10*f[i-1]%mod;
}
int n,k,a[maxn],s[maxn];
char c[maxn];
int C(int n,int m)
{
    if(n<m||m<0)return 0;
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int main()
{
    init();
    while(~scanf("%d%d",&n,&k))
    {
        scanf("%s",c);
        for(int i=0;i<n;i++)a[i]=c[i]-'0',s[i]=(i?s[i-1]:0)+a[i];
        int ans=0;
        for(int i=0;i<=n-2;i++)add(ans,(ll)s[n-2-i]*f[i]%mod*C(n-2-i,k-1)%mod);
        for(int i=0;i<n;i++)add(ans,(ll)a[i]*f[n-1-i]%mod*C(i,k)%mod);
        printf("%d\n",ans);
    }   
    return 0;
}