oj.uz

Submission #229936

# Submission time Handle Problem Language Result Execution time Memory
229936 2020-05-07T07:33:40 Z mohamedsobhi777 Rima (COCI17_rima) C++14
56 / 140
 338 ms 250016 KB 
#pragma comment(linker, "/STACK: 3000000")

#include<bits/stdc++.h>

using namespace std ; 


const int N = 3e6 + 7 ; 

int n  ; 
string ss[N] ; 

struct trie{
    int nxt[N][27] ; 
    int cnt[N] ; 
    int t = 1 ; 
    int dp[N][2] ; 
    void add(string &add_ ){
        int cur = 1 ; 
        int sz = add_.size() ; 
        for(int i = 0 ;i < sz ;i++ ){
            int c = add_[i] - 'a' ; 
            if(!nxt[cur][c])nxt[cur][c] = ++t ;
            cur = nxt[cur][c] ;
        }
        cnt[cur] ++ ; 
    }

    int solve(int i , bool flag){
        if(dp[i][flag]!=-1) return dp[i][flag] ;
        if(flag && !cnt[i]) return 0 ; 
        int ret = 0 ; 
        int kids = 0 ;  
        
        for(int j = 0 ;j < 26 ;j ++){
            kids +=cnt[nxt[i][j]] ; 
        }
        for(int j = 0 ;j < 26 ; j++){
            if(nxt[i][j]){
                ret = max(ret ,  solve(nxt[i][j] , flag )  ) ; 
                ret = max(ret , solve(nxt[i][j] , 1) + kids ) ; 
            }
        }   
        return dp[i][flag] =ret ; 
    }

} tr; 

int main(){
    ios_base::sync_with_stdio(0) ; 
    cin.tie(0) ; 
    //freopen("in.in" , "r" , stdin) ; 
    memset(tr.dp , -1 , sizeof tr.dp) ; 
    cin>>n ; 
    for(int i = 0 ; i < n ;i++){
        string x ; 
        cin>>x ; 
        reverse( x .begin() , x.end()) ; 
        tr.add(x) ; 
    }
    int ans = max(tr.solve(1 , 0 ) , tr.solve(1 , 1)) ; 
    cout<< ans;  
    return 0 ;
}

Compilation message

rima.cpp:1:0: warning: ignoring #pragma comment  [-Wunknown-pragmas]
 #pragma comment(linker, "/STACK: 3000000")

Subtask #1
56.0 / 140.0

# Verdict Execution time Memory Grader output
1 Correct 68 ms 117880 KB Output is correct
2 Correct 68 ms 117880 KB Output is correct
3 Correct 68 ms 117884 KB Output is correct
4 Incorrect 338 ms 185056 KB Output isn't correct
5 Correct 89 ms 118776 KB Output is correct
6 Incorrect 167 ms 248508 KB Output isn't correct
7 Incorrect 165 ms 248420 KB Output isn't correct
8 Incorrect 166 ms 248352 KB Output isn't correct
9 Incorrect 186 ms 250016 KB Output isn't correct
10 Incorrect 164 ms 248524 KB Output isn't correct