Submission #245399

#	Time	Username	Problem	Language	Result	Execution time	Memory
245399		leaked	Rima (COCI17_rima)	C++14	140 / 140	418 ms	159992 KiB

rima

#include<bits/stdc++.h>
using namespace std;
//#include<ext/pb_ds/assoc_container.hpp>
//#include<ext/pb_ds/tree_policy.hpp>
//
//    #pragma GCC optimize("unroll-loops")
//    #pragma GCC optimize("Ofast")
//    #pragma GCC optimize("-O3")
//    #pragma GCC optimize("no-stack-protector")
//    #pragma GCC optimize("fast-math")
//#define LOCAL
#define sim template < class c
#define ris return * this
#define dor > debug & operator <<
#define eni(x) sim > typename \
  enable_if<sizeof dud<c>(0) x 1, debug&>::type operator<<(c i) {
sim > struct rge { c b, e; };
sim > rge<c> range(c i, c j) { return rge<c>{i, j}; }
sim > auto dud(c* x) -> decltype(cout<< *x, 0);
sim > char dud(...);
struct debug {
#ifndef LOCAL
~debug() { cout << endl; }
eni(!=) cout << boolalpha << i; ris; }
eni(==) ris << range(begin(i), end(i)); }
sim, class b dor(pair < b, c > d) {
  ris << "(" << d.first << ", " << d.second << ")";
}
sim dor(rge<c> d) {
  *this << "[";
  for (auto it = d.b; it != d.e; ++it)
	*this << ", " + 2 * (it == d.b) << *it;
  ris << "]";
}
#else
sim dor(const c&) { ris; }
#endif
};
#define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "
#define fi first
#define f first
#define se second
#define s second
#define vi_a vector<int>a;
#define p_b push_back
////////////////////////////////???????????????CHECK THIS OUT???????????????//////////////////////////////
#define ll long long
////////////////////////////////???????????????CHECK THIS OUT???????????????//////////////////////////////
#define ld long double
#define pll pair<ll,ll>
#define pii pair<int,int>
#define m_p make_pair
#define fast_io cin.tie(0);cout.tie(0);ios_base::sync_with_stdio(0);
#define all(x) x.begin(),x.end()
#define getfiles    ifstream cin("input.txt");ofstream cout("output.txt");
#define pw(x) (1ll << x)
#define sz(x) (ll)x.size()

#define rall(x) x.rbegin(),x.rend()
#define len(a) (ll)a.size()
#define rep(x,l,r) for(ll x=l;x<r;x++)

//using namespace __gnu_pbds;
ld eps = (ld)1 / 1e6;
const ld pi=3.14159265359;
ll inf = 1e18,mod1=1e9+7;
ll sqr(ll a) { return a * a; }
ll qb(ll a) { return a * a * a; }
ll gcd(ll a, ll b) { return !a ? b : gcd(b % a, a); }
ll binpow(ll a, ll b, ll mod) { return b ? (b % 2 ? (a * (sqr(binpow(a, b / 2, mod)) % mod)) % mod : sqr(binpow(a, b / 2, mod)) % mod) : 1; }
ll binmult(ll a, ll b, ll mod) { return b ? (b % 2 ? (2 * binmult(a, b / 2, mod) + a) % mod : (2 * binmult(a, b / 2, mod)) % mod) : 0; }
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
const ll R=1e4;
const ll tx[4]={0,0,-1,1};
const ll ty[4]={-1,1,0,0};
const char rev_to[4]={'E','W','N','S'};
const int N=3e6+1;
const int M=1e9+7;
struct node{
    node* to[26];
    ll c1=0,c2=0,ans;
    node(){
        c1=0,c2=0,ans=0;
        for(int i=0;i<26;i++) to[i]=nullptr;
    }
};
node* top=new node();
void add(string &s){
    node * cur=top;
    for(int i=0;i<sz(s)-1;i++){
        if(!cur->to[s[i]-'a']) cur->to[s[i]-'a']=new node();
        cur=cur->to[s[i]-'a'];
    }
    if(!cur->to[s.back()-'a']){
        cur->to[s.back()-'a']=new node();
    }
    cur=cur->to[s.back()-'a'];
    cur->c2++;
    cur->ans++;
}
ll rans=1;
ll go(node* cur){
    ll cnt1=0,cnt2=0;
    ll mx=-1;
    for(int i=0;i<26;i++){
        if(cur->to[i]){
            mx=max(mx,go(cur->to[i]));
            cnt1++;
            if(cur->to[i]->c2) cnt2++;
        }
    }
    if(!cnt1){
        cur->ans=1;
        return 1;
    }
    cur->ans+=max(1ll*0,mx+cnt2-1); //debug()<<imie(cur->ans)<<imie(w)<<imie(mx)<<imie(cnt2);
    rans=max(rans,cur->ans);
    if(cur->c2) return cur->ans;
    else return 0;
}
ll mxx(node *cur){
    ll mx1=0,mx2=0,cnt2=0;
    for(int i=0;i<26;i++){
        if(cur->to[i]){
           mxx(cur->to[i]);
           if(cur->c2){
//                rans=max(rans,mx+cur->to[i]->ans);
               // mx=max(mx,(int)cur->to[i]->ans);
                if(cur->to[i]->ans>mx1){
                    mx2=mx1;
                    mx1=cur->to[i]->ans;
                }
                else if(cur->to[i]->ans>mx2) mx2=cur->to[i]->ans;
                if(cur->to[i]->c2)cnt2++;
           }
        }
    }
//      debug()<<imie(rans)<<imie(mx1)<<imie(w)<<imie(mx2);
   if(cur->c2) rans=max(rans,(ll)mx1+mx2+cnt2-1-(cur==top));
}
signed main(){
    fast_io;
    int n;
    cin>>n;
    vector<string>a(n);
    for(auto &z : a) cin>>z,reverse(all(z)),add(z);
    string p;
    go(top);
    top->c2++;
    mxx(top);
    cout<<rans;
    return 0;
}
///stuff you should look for :
/*

6
abc
bc
c
a
da
cda

	* int overflow, array bounds
	* special cases (n=1?)
	* do smth instead of nothing and stay organized
	* read the conditions carefully
	* WRITE STUFF DOWN
*/

Compilation message (stderr)

rima.cpp: In function 'long long int mxx(node*)':
rima.cpp:140:1: warning: no return statement in function returning non-void [-Wreturn-type]
 }
 ^

• Subtask 1: 140 / 140