Submission #251449

#	Time	Username	Problem	Language	Result	Execution time	Memory
251449		VEGAnn	Klavir (COCI17_klavir)	C++14	16 / 160	183 ms	28028 KiB

klavir

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int md = 998244353;
const int M = 1000100;
int n, m, a[M], po[M], ans[M], pf[M];

int mult(int x, int y) { return (1ll * x * y) % md; }

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 >> n >> m;

    po[0] = 1;

    for (int i = 0; i < m; i++) {
        cin >> a[i];
        po[i + 1] = mult(po[i], n);
    }

    pf[0] = 0;

    for (int i = 1; i < m; i++){
        int j = pf[i - 1];

        while (j > 0 && a[j] != a[i])
            j = pf[j - 1];

        if (a[j] == a[i])
            j++;

        pf[i] = j;
    }

    for (int i = 1; i <= m; i++){
        ans[i] = sum(ans[pf[i - 1]], po[i]);

        cout << ans[i] << " ";
    }

    return 0;
}

• Subtask 1: 16 / 16
• Subtask 2: 0 / 16
• Subtask 3: 0 / 16
• Subtask 4: 0 / 16
• Subtask 5: 0 / 16
• Subtask 6: 0 / 16
• Subtask 7: 0 / 16
• Subtask 8: 0 / 16
• Subtask 9: 0 / 16
• Subtask 10: 0 / 16