oj.uz

Submission #520675

# Submission time Handle Problem Language Result Execution time Memory
520675 2022-01-30T17:37:26 Z kakayoshi Rima (COCI17_rima) C++14
140 / 140
 271 ms 110332 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef pair<int, int> pi;
typedef pair<pair<ll, ll>,ll > pii;
typedef vector <ll> vi;
#define forw(i,a,b) for (ll i=a;i<=(b);i++)
#define forb(i,a,b) for (ll i=a;i>=(b);i--)
#define fast {ios::sync_with_stdio(false); cin.tie(0); }
#define fi first
#define se second
#define pu push
#define puf push_front
#define pb push_back
#define pof pop_front
#define pob pop_back
#define fr front
#define all(a) a.begin(),a.end()
const ll oo=1e18;
const ll mod=1337377;
const int base=31;
const int tx[5]={0,-1,0,1,0};
const int ty[5]={0,0,1,0,-1};
const ll maxN=3e5+5;
const ll maxM=1e6+5;
const ll block=700;
int out, save[maxM][30],dem,n;
bool ed[maxM];
vector <int> edge[maxM];
pi largest[maxM];
void update(int &nhat, int &nhi, int val)
{
    if (val>=nhat)
    {
        nhi=nhat;
        nhat=val;
    }
    else
        if (val>=nhi) nhi=val;
    return;
}
void dfs(int u)
{
    largest[u]={0,0};
    int tmp=0;
    for(auto v:edge[u])
    {
        tmp+=ed[v];
        dfs(v);
        update(largest[u].fi,largest[u].se,largest[v].fi);
    }
    tmp-=(largest[u].fi>0);
    tmp-=(largest[u].se>0);
    out=max(out,largest[u].fi+largest[u].se+tmp+(ed[u]==1));
    //cout<<u<<" "<<largest[u].fi<<" "<<largest[u].se<<" "<<tmp<<endl;
    if (!ed[u]) largest[u]={0,0};
    else largest[u].fi+=1+tmp+(largest[u].se>0);
    //cout<<largest[u].fi<<endl;
    return;
}
void solve()
{
    cin>>n;
    forw(i,1,n)
    {
        string s;cin>>s;
        int id=0;
        reverse(all(s));
        for(auto c:s)
        {
            if (save[id][c-'a']==0)
            {
                save[id][c-'a']=++dem;
                edge[id].pb(dem);
            }
            id=save[id][c-'a'];
        }
        ed[id]=1;
    }
    // build
    dfs(0);
    cout<<out;
}
int main()
{
    fast;
    //freopen("test.inp","r",stdin);
    //freopen("test.out","w",stdout);
    int t=1; //cin>>t;
    while (t--)
        solve();
    return 0;
}

Subtask #1
140.0 / 140.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 23756 KB Output is correct
2 Correct 13 ms 23756 KB Output is correct
3 Correct 17 ms 23808 KB Output is correct
4 Correct 271 ms 110332 KB Output is correct
5 Correct 27 ms 24388 KB Output is correct
6 Correct 66 ms 94548 KB Output is correct
7 Correct 61 ms 94516 KB Output is correct
8 Correct 63 ms 94464 KB Output is correct
9 Correct 95 ms 96248 KB Output is correct
10 Correct 57 ms 94432 KB Output is correct