Submission #1105832

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1105832	2024-10-28T03:27:43 Z	nh0902	Rima (COCI17_rima)	C++14	126 / 140	190 ms	123504 KB

rima

#include <bits/stdc++.h>

using namespace std;

const int N = 500010, M = 3000010;

int n;

string s[N];

vector<pair<char, int>> child[M];
bool is_present[M];
int cur;

void add(string s) {
    int pos = 0;
    for (char c : s) {
        int not_child = true;
        int next = 0;
        for (pair<char, int> p : child[pos]) {
            if (p.first == c) {
                not_child = false;
                next = p.second;
                break;
            }
        }
        if (not_child) {
            cur ++;
            next = cur;
            child[pos].push_back({c, next});
        }
        pos = next;
    }
    is_present[pos] = true;
}

int ans;

int dfs(int u) {
    int max_path = 0;
    int max_path2 = 0;
    int cnt_present_child = 0;
    for (pair<char, int> p : child[u]) {
        max_path2 = max(max_path2, dfs(p.second));
        if (max_path2 > max_path) {
            swap(max_path, max_path2);
        }
        if (is_present[p.second]) {
            cnt_present_child ++;
        }
    }
    //cout << u << " : " << max_path << " , " << max_path2 << "\n";

    if (!is_present[u]) {
        return 0;
    }
    if (cnt_present_child >= 2) {
        ans = max(ans, max_path + max_path2 + cnt_present_child - 1);
    }
    else {
        ans = max(ans, max_path + 1);
    }
    //ans = max(ans, max_path + cnt_present_child);
    //cout << u << " : " << height + height2 + 1 << "\n";
    if (cnt_present_child <= 1) {
        //cout << u << " : " << max_path + 1 << "\n";
        return max_path + 1;

    }
    //cout << u << " : " << max_path + cnt_present_child << "\n";
    return max_path + cnt_present_child;
}

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

    cin >> n;

    for (int i = 1; i <= n; i ++) {
        cin >> s[i];
        reverse(s[i].begin(), s[i].end());
        add(s[i]);

     }

    dfs(0);

    cout << ans;
    return 0;
}

Subtask #1

126.0 / 140.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	16 ms	87120 KB	Output is correct
2	Correct	15 ms	87120 KB	Output is correct
3	Correct	19 ms	87136 KB	Output is correct
4	Incorrect	190 ms	99264 KB	Output isn't correct
5	Correct	42 ms	90476 KB	Output is correct
6	Correct	70 ms	120760 KB	Output is correct
7	Correct	69 ms	120596 KB	Output is correct
8	Correct	82 ms	120456 KB	Output is correct
9	Correct	96 ms	123504 KB	Output is correct
10	Correct	77 ms	120460 KB	Output is correct