Submission #1060870

#	Time	Username	Problem	Language	Result	Execution time	Memory
1060870		sleepntsheep	Copy and Paste 3 (JOI22_copypaste3)	C++17	1 / 100	3104 ms	309988 KiB

copypaste3

#include <iostream>
#include <map>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
using ll = long long;
template <typename T>
using ve = vector<T>;

const int N = 2505;
const ll B = 137, M = 1000000007;

int n, lcp[N][N];
ll dp[N][N], a, b, c;
char s[N];

ll hs[N], bp[N], ibp[N];

ll bpow(ll a, ll b) {
    if (!b) return 1;
    ll r { bpow(a, b / 2) };
    if (b & 1) return r * r % M * a % M;
    return r * r % M;
}

void do_hash() {
    bp[0] = ibp[0] = 1;
    for (int i = 1; i <= n; ++i) bp[i] = bp[i - 1] * B % M, ibp[i] = bpow(bp[i], M - 2);
    for (int i = 1; i <= n; ++i) hs[i] = bp[i] * s[i] % M, hs[i] = (hs[i] + hs[i - 1]) % M;
}

ll get_hash(int l, int r) {
    return ibp[l] * (hs[r] - hs[l - 1] + M) % M;
}

map<ll, ve<int>> oo, op[N];

int main() {
    scanf("%d%s%lld%lld%lld", &n, s + 1, &a, &b, &c);

    do_hash();
    memset(dp, 63, sizeof dp);

    for (int i = 1; i <= n; ++i) {
        lcp[i][i] = n - i + 1;
        lcp[n][i] = lcp[i][n] = (s[i] == s[n]);
        dp[i][i] = a;
        for (int j = i; j <= n; ++j)
            oo[get_hash(i, j)].push_back(i);
    }

    for (int i = 1; i <= n; ++i) {
        for (int j = i; j <= n; ++j) {
            ll x = get_hash(i, j), at = i;
            auto &v = oo[x];
            auto it = lower_bound(v.begin(), v.end(), at);
            while (it != v.end()) {
                op[i][x].push_back(*it);
                at += (j - i + 1);
                it = lower_bound(v.begin(), v.end(), at);
            }
        }
    }

    for (int i = n - 1; i >= 1; --i) {
        for (int j = n - 1; j >= 1; --j) {
            lcp[i][j] = max(lcp[i][j], lcp[i + 1][j + 1] + (s[i] == s[j]));
        }
    }


    for (int l = 2; l <= n; ++l) {
        for (int i = 1, j = l; j <= n; ++i, ++j) {
            for (int k = i; k <= j; ++k) {
                int len = k - i + 1;
                ll x = get_hash(i, k);
                auto &oc = op[i][x];
                int cnt = upper_bound(oc.begin(), oc.end(), j - len + 1) - lower_bound(oc.begin(), oc.end(), i);
                int lef = j - i + 1 - cnt * len;
                dp[i][j] = min(dp[i][j], dp[i][k] + lef * a + cnt * c + b);
            }
            dp[i][j] = min(dp[i][j], dp[i + 1][j] + a);
        }
    }

    printf("%lld", dp[1][n]);
}

Compilation message (stderr)

copypaste3.cpp: In function 'int main()':
copypaste3.cpp:40:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   40 |     scanf("%d%s%lld%lld%lld", &n, s + 1, &a, &b, &c);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 1 / 1
• Subtask 2: 0 / 5
• Subtask 3: 0 / 14
• Subtask 4: 0 / 10
• Subtask 5: 0 / 32
• Subtask 6: 0 / 38