結果

問題 No.346 チワワ数え上げ問題
コンテスト
ユーザー Naco
提出日時 2019-07-12 15:48:49
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,232 bytes
コンパイル時間 495 ms
コンパイル使用メモリ 61,088 KB
実行使用メモリ 8,124 KB
最終ジャッジ日時 2024-11-17 15:43:56
合計ジャッジ時間 2,257 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<map>
using namespace std;
const int MAX=210000;
const long long MOD=1000000007;

typedef long long ll;

ll fac[MAX],finv[MAX],inv[MAX];

ll mod_pow(ll x, ll n){ //べき乗
    ll ans=1;
    while(n!=0){
        if(n&1) ans=ans*x%MOD;
        x=x*x%MOD;
        n=n>>1;
    }
    return ans;
}

void com_init(){ //初期化
    fac[0]=fac[1]=1;
    finv[0]=finv[1]=1;
    inv[1]=1;
    for(int i=2;i<MAX;i++){
        fac[i]=fac[i-1]*i%MOD;
        //inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD;
        //finv[i]=finv[i-1]*inv[i]%MOD;
        finv[i]=finv[i-1]*mod_pow(i,MOD-2)%MOD;//フェルマーの小定理より
    }
}

ll comb(ll n, ll k){ //nCk
    if(n<k) return 0;
    if(n<0||k<0) return 0;
    return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD;
}

int main(){
    string s;
    cin >> s;
    int imos[s.size()];
    for(int i=0;i<s.size();i++){
        imos[i]=0;
    }
    for(int i=0;i<s.size();i++){
        if(s[i]=='w') imos[i]++;
    }
    for(int i=1;i<s.size();i++){
        imos[i]+=imos[i-1];
    }
    com_init();
    ll ans=0;
    for(int i=0;i<s.size();i++){
        if(s[i]=='c'){
            ll w=imos[s.size()-1]-imos[i];
            ans+=comb(w,2);
        }
    }
    cout << ans << endl;
}
0