Codeforces 521C 組合數取模(乘法逆元)

【題目鏈接】
http://acm.hust.edu.cn/vjudge/contest/view.action?cid=106914#overview

【解題報告】
之前很少遇到組合數取模的問題(做題太少了),所以就GG了……組合數取模這一問題在算法競賽中還是很常見的,必須紮實掌握。
回到這道題目來,你需要在n個數之間放k個加號,然後求出所有方案的和。
我們知道正向思維,即求出所有的方案,然後對每個方案進行求和是不可取的(數據規模太大)。
所以把這個過程去冗餘,怎麼去冗餘呢?我們知道位置i的數字a[i]會在C(n-1,k)種方案中出現,而它在每種方案中只可能以第1位(從低位開始計),第二位…..這些情況出現。
所以我們考慮枚舉每一位數字在各個位置上出現的次數

設s[i]表示第i位數字a[i]在總和中的貢獻,
s[i]/a[i]=10^(i-1)C(n-i,k)+ sigma{   10^(j-1)C(n-j-1,k-1)    }( 1<=j<=i-1 )

我們直接對這個式子進行求和即可。(要注意,需要把右面的式子提取出來,處理一下前綴和,否則O(n^2)的時間複雜度是無法承受的)
由於n的範圍爲n<=1e5,所以這裏的組合數取模需要使用乘法逆元。
這裏學習了一個通過預處理O(1)查詢C(n,m)的方法。
其中fac[m]是m!; inv[m]是m!%mod的逆元。
這裏寫圖片描述

【參考代碼】

#include<bits/stdc++.h>
using namespace  std;
typedef long long LL;
const int maxn=3e5+100;
const int mod=1e9+7;
int n,k;
char s[maxn];
LL a[maxn];


LL Pow[maxn],fac[maxn],inv[maxn],f[maxn];
void init()
{
      scanf( "%d%d%s",&n,&k,s+1 );
      Pow[0]=1;
      for( int i=1;i<=n;i++ ) Pow[i]=Pow[i-1]*10%mod;
      for(  int i=1;i<=n;i++ ) a[i]=s[n-i+1]-'0';

      fac[0]=fac[1]=inv[0]=inv[1]=f[0]=f[1]=1;
      for(  int i=2;i<maxn;i++ )
      {
            fac[i]=fac[i-1]*i%mod;
            LL t=mod/i,k=mod%i;
            f[i]=(mod-t)*f[k]%mod;
            inv[i]=inv[i-1]*f[i]%mod;
      }
}

LL C( int n, int m  )
{
      if( m<0 || m>n )return 0;
      return fac[n] * inv[m] %mod * inv[n-m] % mod;
}

LL dp[maxn];
void solve()
{
      dp[0]=0;
      for(  int j=1;j<=n-1;j++ )
      {
            dp[j]=dp[j-1]+Pow[j-1]*C(  n-j-1,k-1 );
            dp[j]%=mod;
      }
}


int main()
{
      init();
      LL ans=0;
      solve();
      for( int i=1;i<=n;i++ )
      {
            ans+=a[i]*dp[i-1];
            ans%=mod;
            ans+=a[i]*Pow[i-1]*C(n-i,k);
            ans%=mod;
      }
      cout<<ans<<endl;
      return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章