ABC167 E - Colorful Blocks（思维+组合数学）

题意：

解法：

枚举相邻的位置有i个,可以视作从n-1个(k,k+1)二元组中选出i个出来,方案数为C(n-1,i).
那么剩下的n-i个位置一定不相同,方案数为m*((m-1)^(n-i-1)).

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
const int mod=998244353;
int fac[maxm],inv[maxm];
int n,m,k;
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init(){
    fac[0]=1;
    for(int i=1;i<maxm;i++)fac[i]=fac[i-1]*i%mod;
    inv[maxm-1]=ppow(fac[maxm-1],mod-2,mod);
    for(int i=maxm-2;i>=0;i--)inv[i]=(i+1)*inv[i+1]%mod;
}
int C(int n,int m){
    if(m<0||m>n)return 0;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
void solve(){
    init();
    cin>>n>>m>>k;
    int ans=0;
    for(int i=0;i<=k;i++){//枚举颜色相同的相邻位置个数
        int t=C(n-1,i);
        int tt=m*ppow(m-1,n-i-1,mod)%mod;
        ans=(ans+t*tt%mod)%mod;
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---