Submission #994106

# Submission time Handle Problem Language Result Execution time Memory
994106 2024-06-07T06:29:52 Z vjudge1 Rima (COCI17_rima) C++17
56 / 140
228 ms 148668 KB
#include <bits/stdc++.h>
using namespace std;

const int N = 5e5 + 10;
int n, sz;
string S[N];
int dp[N * 10];
bool have[N * 10];
vector<pair<char, int>> g[N * 10];

void add(string s){
    int cur = 0;
    for (int i = 0; i < s.size(); i ++){
        int nxt = 0;
        for (auto [c, v] : g[cur]){
            if (c == s[i]){
                nxt = v;
                break;
            }
        }

        if (nxt){
            cur = nxt;
        }
        else{
            for (int j = i; j < s.size(); j ++){
                sz++;
                g[cur].push_back({s[j], sz});
                cur = sz;
            }
            break;
        }
    }

    have[cur] = 1;
}

void print(){
    for (int i = 0; i <= sz; i ++){
        cout << "current node = " << i << endl;
        for (auto [c, v] : g[i]){
            cout << "child " << v << " with char " << c << endl; 
        }
    }
}

int main(){
    cin >> n;
    for (int i = 0; i < n; i ++){
        cin >> S[i];
        reverse(S[i].begin(), S[i].end());
        add(S[i]);
    }

    int ans = 0;
    // print();
    // cout << endl;
    for (int i = sz; i >= 0; i --){
        // cout << "current node = " << i << endl;

        int sm = 0;
        int mx = 0;
        for (auto [c, v] : g[i]){
            // cout << "going to " << v << " with char " << c << endl;
            if (have[v]){
                sm ++;
                mx = max(mx, dp[v]);
            }
        }
        if (sm == 0)
            dp[i] = 0;
        else 
            dp[i] = sm + mx;
        ans = max(ans, dp[i]);
        // cout << "Answer = " << dp[i] << endl;
    }

    cout << ans << endl;
}

Compilation message

rima.cpp: In function 'void add(std::string)':
rima.cpp:13:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |     for (int i = 0; i < s.size(); i ++){
      |                     ~~^~~~~~~~~~
rima.cpp:26:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   26 |             for (int j = i; j < s.size(); j ++){
      |                             ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 55 ms 133340 KB Output is correct
2 Correct 60 ms 133396 KB Output is correct
3 Correct 54 ms 133456 KB Output is correct
4 Incorrect 228 ms 147868 KB Output isn't correct
5 Correct 106 ms 137812 KB Output is correct
6 Incorrect 85 ms 144840 KB Output isn't correct
7 Incorrect 79 ms 144584 KB Output isn't correct
8 Incorrect 75 ms 144600 KB Output isn't correct
9 Incorrect 131 ms 148668 KB Output isn't correct
10 Incorrect 73 ms 144524 KB Output isn't correct