Submission #1133

#	Time	Username	Problem	Language	Result	Execution time	Memory
1133		tncks0121	백신 (KOI13_vaccine)	C++98	13.68 / 24	1000 ms	219864 KiB

vaccine

#include <stdio.h> 
#include <stdlib.h> 
#include <string.h> 
#include <memory.h> 
#include <math.h> 
#include <assert.h> 
#include <stack> 
#include <queue> 
#include <map> 
#include <set> 
#include <algorithm> 
#include <string> 
#include <functional> 
#include <vector> 
#include <deque> 
#include <utility> 
#include <bitset> 
#include <limits.h>  

using namespace std; 
typedef long long ll; 
typedef unsigned long long llu; 
const int INF = 987654321; 
const ll LINF = 1e15;  

const int N_ = 1005, M_ = 1005;
const int SN_ = 2002005;
const int lgSN_ = 21;

int N, K, A[M_];
int S[SN_], X[SN_], SN;

int SA[SN_], SS[SN_], Ord[lgSN_][SN_];
int L, lgL;

inline bool cmpSA (const int &a, const int &b) {
    if(Ord[lgL][a] != Ord[lgL][b]) return Ord[lgL][a] < Ord[lgL][b];
    int ta = 0, tb = 0;
    if(a + L <= SN) ta = Ord[lgL][a + L];
    if(b + L <= SN) tb = Ord[lgL][b + L];
    return ta < tb;
}

inline int LCP (int x, int y) {
    int r = 0;
    for(int i = lgL; --i; ) {
        if(Ord[i][x+r] == Ord[i][y+r]) r |= 1<<i;
    }
    return r;
}

int chk[N_];

int start[SN_], lnk[SN_], val[SN_];

int main() {
    
    int i, j, k;
    
    scanf("%d%d", &N, &K);N*=10;
    for(i = 1; i <= N; i+=10) {
        int mi; scanf("%d", &mi);
        for(j = 1; j <= mi; j++) scanf("%d", A+j);
      for(k=0; k<10; k++){
        for(j = 1; j <= mi; j++) ++SN, S[SN] = A[j], X[SN] = i+k; S[++SN] = 0;
        for(j = mi; j >= 1; j--) ++SN, S[SN] = A[j], X[SN] = i+k; S[++SN] = 0;
      }
    }
    
    for(i = 1; i <= SN; i++) SS[i] = SS[i - 1] + (S[i] == 0), SA[i] = i;
    for(i = 1; i <= SN; i++) Ord[0][i] = S[i];
    
    for(L = 1, lgL = 0; L <= SN; L <<= 1, lgL++) {
        for(k = L; k >= 0; k -= L) {
            for(i = 1; i <= SN; i++) {
                int v = 0; if(SA[i] + k <= SN) v = Ord[lgL][SA[i] + k];
                lnk[i] = start[v]; start[v] = i; 
                val[i] = SA[i];
            }
            int sn = SN;
            i = SN; if(L == 1) i = 10000;
            for(; i >= 0; i--) {
                for(j = start[i]; j > 0; j = lnk[j]) {
                    SA[sn--] = val[j];
                    val[j] = 0;
                }
                start[i] = 0;
            }
        }
      
        Ord[lgL + 1][SA[1]] = 1;
        for(i = 2; i <= SN; i++) {
            Ord[lgL + 1][SA[i]] = Ord[lgL + 1][SA[i - 1]];
            if(cmpSA(SA[i - 1], SA[i])) ++Ord[lgL + 1][SA[i]];
        }
    }
    
    for(i = 1; i <= SN; i = j) {
        int count = 1; if(X[SA[i]] == 0) count = 0;
        chk[X[SA[i]]] = i;
        for(j = i + 1; j <= SN; j++) {
            if(LCP(SA[j - 1], SA[j]) < K) break;
            if(chk[X[SA[j]]] != i && X[SA[j]] > 0) ++count;
            chk[X[SA[j]]] = i;
        }
        if(count == N) {
            if(SS[SA[i] + K - 1] - SS[SA[i] - 1] > 0) continue;
            puts("YES");
            return 0;
        }
    }
    
    puts("NO");
    return 0;
}

• Subtask 1: 2.64 / 2.64
• Subtask 2: 3.12 / 3.12
• Subtask 3: 3.6 / 3.6
• Subtask 4: 4.32 / 4.32
• Subtask 5: 0 / 4.8
• Subtask 6: 0 / 5.52