oj.uz

답안 #239285

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
239285 2020-06-15T08:21:00 Z topovik Mate (COCI18_mate) C++14
100 / 100
 1561 ms 13692 KB 
#include <bits/stdc++.h>

#pragma GCC optimize("O3")
#define f first
#define s second
#define pb push_back
#define INF 1000000000
#define N (long)1e3*5+2

using namespace std;

typedef long long ll;
typedef long double ld;

const ll oo = 1e9+7;

int sf[26][N];

int otr[N][N];

int dp[26][26][N];

long long fact[N];
long long inv[N];
long long invfact[N];

inline long long C(ll n, ll k) {
	return fact[n] * invfact[k] % oo * invfact[n - k] % oo;
}

int main()
{
    ios_base::sync_with_stdio(0);
    iostream::sync_with_stdio(0);
    cin.tie(0);
    fact[0] = invfact[0] = 1;
	for (ll i = 1; i < N; ++i) {
		inv[i] = (i == 1) ? 1 : oo - oo / i * inv[oo % i] % oo;
		fact[i] = fact[i - 1] * i % oo;
		invfact[i] = invfact[i - 1] * inv[i] % oo;
		assert(inv[i] * i % oo == 1);
	}
    string s;
    cin>>s;
    ll n=s.size();
    for (ll i=0; i<26; i++)
    {
        sf[i][n]=0;
        for (ll j=n-1; j>=0; j--) sf[i][j]=sf[i][j+1]+((s[j]-'a')==i);
    }
    for (register ll i=0; i<n; i++)
    {
        ll x=s[i]-'a';
        for (register ll j=0; j<26; j++)
        {
            const register int m=(char)(j+'a')==s[i]?sf[j][i+1]:sf[j][i];
            for (ll c=0; c<=i; c++) dp[x][j][c+2]=(dp[x][j][c+2]+(C(i,c)*m)%oo)%oo;
        }
    }
    ll q;
    cin>>q;
    while (q>0)
    {
        q--;
        ll x;
        char ch,ch1;
        cin>>x>>ch>>ch1;
        cout<<dp[ch-'a'][ch1-'a'][x]<<endl;
    }
}

Subtask #1
100.0 / 100.0

# 결과 실행 시간 메모리 Grader output
1 Correct 59 ms 3320 KB Output is correct
2 Correct 44 ms 2944 KB Output is correct
3 Correct 46 ms 3072 KB Output is correct
4 Correct 78 ms 3064 KB Output is correct
5 Correct 268 ms 4728 KB Output is correct
6 Correct 313 ms 4732 KB Output is correct
7 Correct 233 ms 4728 KB Output is correct
8 Correct 204 ms 4472 KB Output is correct
9 Correct 1555 ms 13656 KB Output is correct
10 Correct 1561 ms 13692 KB Output is correct