結果
問題 | No.346 チワワ数え上げ問題 |
ユーザー |
|
提出日時 | 2019-10-08 13:08:58 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,134 bytes |
コンパイル時間 | 1,574 ms |
コンパイル使用メモリ | 162,520 KB |
実行使用メモリ | 16,412 KB |
最終ジャッジ日時 | 2024-11-06 22:12:39 |
合計ジャッジ時間 | 4,613 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 WA * 2 |
ソースコード
#include "bits/stdc++.h" using namespace std; typedef long long int ll; const int MAX = 510000; const int MOD = 1000000007; long long fac[MAX], finv[MAX], inv[MAX]; // テーブルを作る前処理 void COMinit() { 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; } } // 二項係数計算 long long COM(int n, int k){ if (n < k) return 0; if (n < 0 || k < 0) return 0; return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD; } int main(){ //階乗を前計算 COMinit(); string S; cin >> S; vector<ll > w; for(ll i = 0; i < S.length(); i++){ if(S[i] == 'c'){ w.push_back(0); }else if(w.size() > 0 && S[i] == 'w'){ for(ll j = 0; j < w.size(); j++){ w[j]++; } } } ll ans = 0; for(ll j = 0; j < w.size(); j++){ //cout << w[j] << endl; ans += COM(w[j], 2); } cout << ans << endl; }