oj.uz

Submission #632209

# Submission time Handle Problem Language Result Execution time Memory
632209 2022-08-19T16:17:23 Z AA_Surely Dabbeh (INOI20_dabbeh) C++14
0 / 100
 1768 ms 96836 KB 
#pragma GCC optimize("Ofast,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")

#include <bits/stdc++.h>

#define FOR(i, x, n) for(int i = x; i < n; i++)
#define F0R(i, n) FOR(i, 0, n)
#define ROF(i, x, n) for(int i = n - 1; i >= x; i--)
#define R0F(i, n) ROF(i, 0, n)

#define WTF cout << "WTF" << endl

#define IOS ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define F first
#define S second
#define PB push_back

#define ALL(x) x.begin(), x.end()
#define RALL(x) x.rbegin(), x.rend()

using namespace std;

typedef long long LL;

typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;

typedef vector<int> VI;
typedef vector<LL> VLL;
typedef vector<PII> VPII;
typedef vector<PLL> VPLL;

const int Q = 1e4 + 7;
const int N = 5e5 + 7;
const int L = 1e6 + 7;
const int MOD = 1e9 + 7;
const int LOG = 21;

#define endl '\n'
#define lc now << 1
#define rc now << 1 | 1

int q, m, n, pw, l;
int tmp[N], mid[L];
int rnk[L][LOG];
PII p[L];
string ts[Q], s, ns;

struct IntMaxSeg {
    int tree[N << 2];

    int build(int now = 1, int ls = 0, int rs = n) {
        if (ls == rs) return tree[now] = tmp[ls];
        int mid = (ls + rs) >> 1;
        return tree[now] = max(build(lc, ls, mid), build(rc, mid + 1, rs));
    }

    int get(int lq, int rq, int now = 1, int ls = 0, int rs = n) {
        if (rq < lq || rq < ls || rs < lq) return 0;
        if (lq <= ls && rs <= rq) return tree[now];
        int mid = (ls + rs) >> 1;
        return max(get(lq, rq, lc, ls, mid), get(lq, rq, rc, mid + 1, rs));
    }
} dp[LOG];

void init() {
    cin >> q >> m;
    F0R(i, q) {
        cin >> ts[i];
        ns += "#" + ts[i];
    }

    cin >> s;
    n = s.length();

    ns += "#" + s;
    l = ns.length();
}

inline bool cmp(const PII &a, const PII &b) {
    if (rnk[pw - 1][a.F] != rnk[pw - 1][b.F]) return rnk[pw - 1][a.F] < rnk[pw - 1][b.F];
    if (max(a.F, b.F) + (1 << pw) >= l) return a.F > b.F;
    return rnk[pw - 1][a.F + (1 << (pw - 1))] < rnk[pw - 1][b.F + (1 << (pw - 1))];
}

void buildSA() {
    F0R(i, l) rnk[0][i] = ns[i];
    int c = -1;
    FOR(i, 1, l) p[i] = {i, (ns[i - 1] == '#' ? c-- : 0)};

    c = n;
    R0F(i, l) {
        if (ns[ p[i].F ] == '#') break;
        p[i].S = c--;
    }

    for(pw = 1; pw < LOG; pw++) {
        sort(p, p + l, cmp);

        rnk[pw][ p[0].F ] = 0;
        FOR(i, 1, l) rnk[pw][ p[i].F ] = rnk[pw][ p[i - 1].F ] + cmp(p[i - 1], p[i]);

        /*cout << "now pw = " << pw << endl << " ranks are ";
        F0R(i, l) cout << rnk[pw][i] << ' '; cout << endl;*/
        
    }
    pw--;
    
    return;
}

inline int lcp(int a, int b) {
    int ret = 0;
    R0F(i, LOG) {
        if (max(a, b) + (1 << i) > l) continue;
        if (rnk[i][a] != rnk[i][b]) continue;

        ret |= (1 << i);
        a += (1 << i);
        b += (1 << i);
    }

    return ret;
}

void dpBase() {
    //F0R(i, l) cout << p[i].F << ' '; cout << endl; cout << "ns = " << ns << endl;
    FOR(i, 1, l) mid[i] = lcp(p[i - 1].F, p[i].F);
    //cout << "lcps = "; FOR(i, 1, l) cout << mid[i] << ' '; cout << endl;

    int mx = 0;
    FOR(i, 1, l) {
        mx = min(mx, mid[i]);
        if (p[i].S < 0) 
            mx = ts[ -p[i].S - 1 ].length();
        if (p[i].S > 0) 
            tmp[ p[i].S - 1 ] = mx;
    }

    mx = 0;
    R0F(i, l) {
        if (i < l - 1) mx = min(mx, mid[i + 1]);
        if (p[i].S < 0) 
            mx = ts[ -p[i].S - 1 ].length();
        if (p[i].S > 0) 
            tmp[ p[i].S - 1 ] = max(tmp[ p[i].S - 1 ], mx);
    }

    F0R(i, n) tmp[i] += i;
    //F0R(i, n) cout << tmp[i] << ' '; cout << endl;

    dp[0].build();
}

void calcLayer(int id) {
    F0R(i, n + 1) {
        int x = dp[id - 1].get(i, i);
        tmp[i] = dp[id - 1].get(i, x);
    }

    dp[id].build();
}

void handleQuery() {
    int l, r; 
    cin >> l >> r;
    r--;

    int now = l, sum = 0;
    if (dp[LOG - 1].get(l, l) <= r) {
        cout << -1 << endl;
        return;
    }

    R0F(i, LOG) {
        int x = dp[i].get(l, now);
        if (x <= r) {
            sum += (1 << i);
            now = x;
        }
    }
    
    cout << sum + 1 << endl;
}

int main() {
    IOS;

    init();
    buildSA();

    dpBase();

    FOR(i, 1, LOG) calcLayer(i);
    while(m--) handleQuery();

    return 0;
}

Subtask #1
0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 724 KB Output is correct
2 Incorrect 723 ms 8604 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Incorrect 1768 ms 96836 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 50.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 724 KB Output is correct
2 Incorrect 723 ms 8604 KB Output isn't correct
3 Halted 0 ms 0 KB -