結果

問題 No.1171 Runs in Subsequences
ユーザー pes_magic
提出日時 2020-08-14 22:49:55
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 1,303 bytes
コンパイル時間 883 ms
コンパイル使用メモリ 75,648 KB
実行使用メモリ 11,016 KB
最終ジャッジ日時 2024-10-10 16:11:14
合計ジャッジ時間 1,586 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>

using namespace std;

const int MOD = 1000000007;

long long solve(const string& S){
    const int N = S.size();
    const int maxSize = N+1;
	vector<long long> inv(maxSize);
	vector<long long> fact(maxSize);
	vector<long long> factInv(maxSize);
	for(int i=0;i<2;i++) inv[i] = fact[i] = factInv[i] = 1;
	for(int i=2;i<maxSize;i++){
		inv[i] = inv[MOD % i] * (MOD - MOD / i) % MOD;
		fact[i] = fact[i-1] * i % MOD;
		factInv[i] = factInv[i-1] * inv[i] % MOD;
	}
	auto comb = [&](int n, int r){
		if(n < r || r < 0) return 0LL;
		return fact[n] * factInv[n-r] % MOD * factInv[r] % MOD;
	};
    vector<vector<int>> pos(26);
    vector<long long> pow(N+1, 1);
    for(int i=1;i<=N;i++) pow[i] = 2*pow[i-1]%MOD;
    long long res = 0;
    for(int i=1;i<=N;i++) res = (res + i * comb(N, i)) % MOD;
    for(int i=0;i<N;i++) pos[S[i]-'a'].push_back(i);
    for(auto& v : pos){
        if(v.size() < 2) continue;
        long long base = pow[v[0]];
        for(int i=1;i<v.size();i++){
            long long sub = (base * pow[N-v[i]-1]) % MOD;
            res = (res + MOD - sub) % MOD;
            base = (base + pow[v[i]]) % MOD;
        }
    }
    return res;
}

int main(){
    string S;
    while(cin >> S){
        cout << solve(S) << endl;
    }
}
0