Submission #239028

#	Time	Username	Problem	Language	Result	Execution time	Memory
239028		VEGAnn	Mate (COCI18_mate)	C++14	100 / 100	1135 ms	21500 KiB

mate

#include <bits/stdc++.h>
//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")
#define i2 array<int,2>
#define PB push_back
#define all(x) x.begin(),x.end()
#define sz(x) ((int)x.size())
using namespace std;
typedef long long ll;
const int N = 2010;
const ll OO = 1e18;
const int md = int(1e9) + 7;
vector<int> ps[26];
string s, st;
int n, C[N][N];
ll cnt[N][26], ans;

int sum(int x, int y){
    x += y;
    if (x >= md)
        x -= md;
    return x;
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);

#ifdef _LOCAL
    freopen("in.txt","r",stdin);
#endif // _LOCAL

    cin >> s;
    n = sz(s);

    for (int i = 0; i < n; i++) {
        ps[s[i] - 'a'].PB(i);

        for (int j = i + 1; j < n; j++)
            cnt[i][s[j] - 'a']++;
    }

    C[0][0] = 1;

    for (int i = 1; i <= n; i++){
        C[i][0] = C[i][i] = 1;

        for (int j = 1; j < n; j++)
            C[i][j] = sum(C[i - 1][j], C[i - 1][j - 1]);
    }

    int qq; cin >> qq;

    for (; qq; qq--){
        int d; cin >> d >> st;

        d -= 2;

        int ch = (st[1] - 'a'), C1 = (st[0] - 'a');

        ans = 0;

        for (int i : ps[C1]) {
            if (cnt[i][ch] == 0) break;

            (ans += cnt[i][ch] * C[i][d]) %= md;
        }

        cout << ans << '\n';
    }

    return 0;
}

• Subtask 1: 100 / 100