Submission #542519

# Submission time Handle Problem Language Result Execution time Memory
542519 2022-03-26T20:05:26 Z Olympia Savez (COCI15_savez) C++17
0 / 120
85 ms 41024 KB
#include <vector>
#include <algorithm>
#include <iostream>
#include <set>
#include <cmath>
#include <map>
#include <random>
#include <cassert>
#include <ctime>
#include <cstdlib>
#include <queue>
#include <limits.h>
using namespace std;

int64_t binPow (int64_t x, int64_t y, int64_t MOD) {
    x %= MOD;
    int64_t res = x;
    int64_t ans = 1;
    while (y > 0) {
        if (y & 1) {
            ans *= res, ans %= MOD;
        }
        res *= res, res %= MOD;
        y /= 2;
    }
    return ans;
}
int64_t inv (int64_t x, int64_t MOD) {
    return binPow(x, MOD - 2, MOD);
}
vector<int64_t> powr = {1};
vector<int64_t> ipowr = {1};
struct Hasher {
    void resz (int mod, int base, string str) {
        int n = str.size();
        this->sz = n;
        this->MOD = mod, this->BASE = base;
        this->pref.assign(n + 1, 0);
        for (int i = 1; i <= n; i++) {
            pref[i] = (pref[i - 1] + (powr[n - i] * (str[i - 1] - 'A' + 1)) % MOD) % MOD;
            pref[i] %= MOD;
        }
    }
    int sz;
    int MOD;
    int BASE;
    vector<int64_t> pref;
    int64_t query (int l, int r) {
        return (ipowr[sz - r - 1] * (pref[r + 1] - pref[l] + MOD) % MOD) % MOD;
    }
};
vector<int> dp, w;
vector<vector<int>> adj;
int myMax = 0;
void rec (int curNode) {
    if (dp[curNode] != 0) {
        return;
    }
    dp[curNode] = w[curNode];
    for (int i: adj[curNode]) {
        rec(i);
        dp[curNode] = max(dp[curNode], dp[i] + w[curNode]);
    }
    myMax = max(myMax, dp[curNode]);
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    while (powr.size() != 1e6 + 5) {
        powr.push_back(powr.back() * 31);
        powr.back() %= (int)1e9 + 9;
    }
    ipowr.resize(powr.size());
    ipowr[0] = 1; ipowr[1] = inv(powr[1], 1e9 + 9);
    for (int i = 2; i < powr.size(); i++) {
        ipowr[i] = (ipowr[i - 1] * ipowr[1]) % ((int)1e9 + 9);
    }
    int N;
    cin >> N;
    vector<string> vec(N);
    for (int i = 0; i < N; i++) {
        cin >> vec[i];
    }
    map<string,int> m;
    map<int64_t,int> myMap;
    int64_t ans = 0;
    adj.resize(N);
    dp.assign(N, 0);
    w.resize(N);
    for (int i = 0; i < N; i++) {
        if (m.count(vec[i])) {
            m[vec[i]]++;
            continue;
        }
        m[vec[i]]++;
        Hasher h1; h1.resz(1e9 + 9, 31, vec[i]);
        for (int l = 1; l <= vec[i].size() - 1; l++) {
            if (h1.query(0, l - 1) == h1.query(vec[i].size() - l, vec[i].size() - 1)) {
                if (myMap.count(h1.query(0, l - 1))) {
                    //cout << vec[i] << " " << l << '\n';
                    adj[i].push_back(myMap[h1.query(0, l - 1)]);
                }
            }
        }
        myMap[h1.query(0, vec[i].size() - 1)] = i;
    }
    for (int i= 0; i < vec.size(); i++) {
        w[i] = m[vec[i]];
    }
    for (int i = 0; i < N; i++) {
        rec(i);
    }
    cout << myMax;


}

Compilation message

savez.cpp: In function 'int main()':
savez.cpp:75:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   75 |     for (int i = 2; i < powr.size(); i++) {
      |                     ~~^~~~~~~~~~~~~
savez.cpp:97:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   97 |         for (int l = 1; l <= vec[i].size() - 1; l++) {
      |                         ~~^~~~~~~~~~~~~~~~~~~~
savez.cpp:107:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::__cxx11::basic_string<char> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  107 |     for (int i= 0; i < vec.size(); i++) {
      |                    ~~^~~~~~~~~~~~
savez.cpp:86:13: warning: unused variable 'ans' [-Wunused-variable]
   86 |     int64_t ans = 0;
      |             ^~~
# Verdict Execution time Memory Grader output
1 Correct 23 ms 16068 KB Output is correct
2 Incorrect 24 ms 16068 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 23 ms 16068 KB Output is correct
2 Correct 24 ms 16148 KB Output is correct
3 Incorrect 26 ms 16232 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 27 ms 17040 KB Output is correct
2 Incorrect 29 ms 18060 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 23 ms 16136 KB Output is correct
2 Correct 45 ms 18656 KB Output is correct
3 Incorrect 60 ms 19080 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 28 ms 17604 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 31 ms 18460 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 37 ms 19852 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 42 ms 21580 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 60 ms 31012 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 85 ms 41024 KB Output isn't correct
2 Halted 0 ms 0 KB -