Submission #244941

#	Time	Username	Problem	Language	Result	Execution time	Memory
244941		NONAME	Rima (COCI17_rima)	C++14	140 / 140	368 ms	150544 KiB

rima

#include <bits/stdc++.h>
#define dbg(x) cerr << #x << " = " << x << "\n"
#define fast_io ios_base::sync_with_stdio(0); cin.tie(0); cout.tie()
using namespace std;
using ll = long long;

const int N = 5e5 + 10;

int n, ans;
string s[N];

struct trie {
    trie *nxt[26];
    int cnt, res;

    trie() {
        for (int i = 0; i < 26; ++i)
            nxt[i] = 0;
        cnt = 0;
        res = 0;
    }

    void add(string &t, int p = 0) {
        if (p == int(t.size())) {
            ++cnt;
            return;
        }

        int j = t[p] - 'a';

        if (nxt[j] == 0)
            nxt[j] = new trie();

        nxt[j] -> add(t, p + 1);
    }

    int f(int x) {
        vector <int> v;
        v.clear();

        int total = 0;

        for (int i = 0; i < 26; ++i) {
            if (nxt[i] == 0)
                continue;

            int cur = (nxt[i] -> f(i));
            total += (nxt[i] -> cnt);

            if (nxt[i] -> cnt)
                v.push_back(cur);
        }

        sort(v.rbegin(), v.rend());

        int mx = 0;

        for (int i = 0; i < min(int(v.size()), 2); ++i)
            mx += v[i];

        //cerr << char(x + 'a') << " " << (v.empty() ? 0 : v[0]) << " " << mx << " " << total << " " << (this -> cnt) << " " << ans << "\n";

        ans = max(ans, mx + total + (this -> cnt));

        return (v.empty() ? 0 : v[0]) + total;
    }
} tr;

int main() {
    fast_io;

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

    tr.f(-1);

    cout << ans << "\n";
}

• Subtask 1: 140 / 140