oj.uz

Submission #769620

# Submission time Handle Problem Language Result Execution time Memory
769620 2023-06-29T22:08:42 Z gg123_pe Sequence (APIO23_sequence) C++17
47 / 100
 305 ms 45420 KB 
#include "sequence.h"
#include <bits/stdc++.h> 
using namespace std;

typedef pair<int,int> ii; 
#define f(i,a,b) for(int i = a; i < b; i++)
#define fa(i,a,b) for(int i = a; i >= b; i--)

const int N = 5e5 + 5; 
const int inf = 1e9; 

int n; 
int bit[N], c[N];
vector <pair<int,int>> pos[N]; 

void upd(int x, int val){
    for(; x < N; x = (x|(x+1))) bit[x] += val; 
}
int sum(int x){
    int res = 0;
    for(; x >= 0; x = (x&(x+1)) - 1) res += bit[x];  
    return res; 
}
int get(int x){
    int l = 1, r = n; 
    while(l < r){
        int m = (l+r)>>1; 
        if(sum(m) >= x) r = m; 
        else l = m+1; 
    }
    return c[l]; 
}
int subtask_2(vector <int> A){
    int ans = 1; 
    f(i,0,n){
        c[A[i]]++; 
        upd(A[i], 1); 

        f(j,i,n){
            int l = j - i + 1; 
            if(l&1){
                ans = max(ans, get((l+1)/2)); 
            }
            else{
                ans = max(ans, get(l/2)); 
                ans = max(ans, get(l/2 + 1)); 
            }
            if(j+1 < n) {
                c[A[j+1]]++; 
                upd(A[j+1], 1); 
            }
        }
        f(j,i,n){
            c[A[j]]--; 
            upd(A[j], -1); 
        }
    }
    return ans;
}
int id(int pos){
    if(pos == 0) return 0; 
    if(c[1] >= pos) return 1; 
    if(c[1] + c[2] >= pos) return 2; 
    return 3; 
}
int res(vector <int> v){
    vector <int> aux, mini; 
    int len = v.size(), ans = 1; 
    f(i,0,len) {
        aux.push_back(v[i] - 2*i); 
        mini.push_back(v[i] - 2*i); 
    }
    fa(i,len-2,0) mini[i] = min(mini[i+1], mini[i]); 

    f(i,0,len-1){
        int ult = aux[i] + 1; // <= ult
        if(mini[i+1] > ult) continue; 
        int l = i+1, r = len-1; 
        while(l < r){
            int m = (l+r+1)>>1; 
            if(mini[m] > ult) r = m-1;  
            else l = m; 
        }
        ans = max(ans, l-i+1); 
    }
    return ans;     
}
int subtask_4(vector <int> A){
    for(int x: A) c[x]++; 
    int p1 = 0, p2 = 0; 
    if(n&1) p1 = (n+1)/2; 
    else p1 = n/2, p2 = p1 + 1; 

    if(id(p1) == 1 or id(p1) == 3) return c[id(p1)]; 
    if(id(p2) == 1 or id(p2) == 3) return c[id(p2)]; 
    if(c[2] > c[1] and c[2] > c[3]) return c[2]; 

    int ans = c[2];
    vector <int> P1, P3; 
    f(i,0,n){
        if(A[i] == 1) P1.push_back(i); 
        if(A[i] == 3) P3.push_back(i); 
    }
    ans = max(ans, res(P1)); 
    ans = max(ans, res(P3)); 
    return ans; 
}
bool is(int l, int r, int ty){
    if(l > r) return 0; 
    if(l != r) return 1; 
    return l%2 == ty; 
}
int subtask_3(vector <int> A){
    int ans = 0, id = 0; 
    while(id < n){
        int c = 0, ini = id+1, comp = A[id]; 
        while(A[id] == comp and id < n) c++, id++; 
        ans = max(ans, c);  
        pos[comp].push_back({ini, ini + c - 1}); 
    } 
    f(i,1,n+1){
        if(pos[i].size() != 2) continue; 
        int l1 = pos[i][0].first, r1 = pos[i][0].second; 
        int l2 = pos[i][1].first, r2 = pos[i][1].second; 
        int C = r1 - l1 + r2 - l2 + 2;
        int B = -(r1 + 1) + (l2 - 1) + 1;
        int A = n - C - B, ty = (B+C)%2; 

        bool fe = 0; 
        if(is(max(0, B-C), min(A, B+C), ty)) fe = 1; 
        if(is(max(0, B-C-1), min(A, B+C-1), 1-ty)) fe = 1; 
        if(fe) ans = max(ans, C);    
    }
    return ans; 
}



struct ST{
    int lazy[4*N]; 
    ii st[4*N];  // min, max
    ii zero = {inf, -inf}; 

    void down(int id){
        lazy[id<<1] += lazy[id], lazy[id<<1|1] += lazy[id]; 
        st[id<<1].first += lazy[id], st[id<<1].second += lazy[id]; 
        st[id<<1|1].first += lazy[id], st[id<<1|1].second += lazy[id]; 
        lazy[id] = 0; 
    }
    ii merge(ii x, ii y){
        return {min(x.first, y.first), max(x.second, y.second)}; 
    }
    void upd(int id, int l, int r, int x, int y, int val){ // [x, y] += val
        if(r < x or y < l) return ; 
        if(x <= l and r <= y){
            lazy[id] += val; 
            st[id].first += val, st[id].second += val; 
            return; 
        }
        down(id); 
        int m = (l+r)>>1; 
        upd(id<<1, l, m, x, y, val), upd(id<<1, m+1, r, x, y, val); 
        st[id] = merge(st[id<<1], st[id<<1|1]); 
    }
    ii query(int id, int l, int r, int x, int y){
        if(r < x or y < l) return zero; 
        if(x <= l and r <= y) return st[id]; 
        down(id); 
        int m = (l+r)>>1; 
        return merge(query(id<<1, l, m, x, y), query(id<<1|1, m+1, r, x, y)); 
    }
    int val(int pos){
        return query(1, 1, n, pos, pos).first; 
    }
}st; 

ii p[N]; 
int a[N]; 

int subtask_5(vector <int> A){
    f(i,1,n+1){
        int x = A[i-1]; 
        if(p[x].first == 0) p[x].first = i; 
        else p[x].second = i; 
    }

    st.upd(1, 1, n, 1, n, 1); 
    f(i,1,n+1) a[i] = 1; 
    
    f(i,1,n+1){
        int x = p[i].first, y = p[i].second; 
        if(y == 0){
            if(x != 0) st.upd(1, 1, n, x, n, -1-a[x]), a[x] = -1; 
            continue; 
        }
        st.upd(1, 1, n, x, n, -a[x]), a[x] = 0; 
        st.upd(1, 1, n, y, n, -a[y]), a[y] = 0; 

        // check
        int SL = st.val(x-1), _SR = st.val(y); 
        int sum = _SR - SL; 
        int miL = 0, maL = 0; 
        int miR = 0, maR = 0; 
        if(x != 1){
            ii C = st.query(1, 1, n, 1, x-1); 
            miL = min(0, SL - C.second); 
            maL = max(0, SL - C.first); 
        }
        if(y != n){
            ii C = st.query(1, 1, n, y+1, n); 
            miR = min(0, C.first - _SR); 
            maR = max(0, C.second - _SR); 
        }
    
        int mi = miL + miR + sum; 
        int ma = maL + maR + sum; 
        
        if(min(2, ma) - max(-2, mi) + 1 > 0) return 2; 
        st.upd(1, 1, n, x, n, -1-a[x]), a[x] = -1; 
        st.upd(1, 1, n, y, n, -1-a[y]), a[y] = -1; 
    }
    return 1; 
}
int sequence(int Ni, vector<int> A) {
    n = Ni; 

    if(n <= 2000) return subtask_2(A); 
    
    int maxi = 0; 
    for(int x: A) maxi = max(maxi, x); 

    if(maxi <= 3) return subtask_4(A); 

    maxi = 0; 
    for(int x: A) c[x]++; 
    f(i,1,n+1) maxi = max(maxi, c[i]); 

    if(maxi <= 2) return subtask_5(A); 
    return subtask_3(A); 
}

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 11 ms 27732 KB Output is correct
3 Correct 11 ms 27724 KB Output is correct
4 Correct 13 ms 27648 KB Output is correct
5 Correct 11 ms 27748 KB Output is correct
6 Correct 10 ms 27732 KB Output is correct
7 Correct 12 ms 27692 KB Output is correct
8 Correct 12 ms 27652 KB Output is correct
9 Correct 13 ms 27648 KB Output is correct
10 Correct 12 ms 27848 KB Output is correct
11 Correct 11 ms 27764 KB Output is correct
12 Correct 11 ms 27660 KB Output is correct

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 11 ms 27732 KB Output is correct
3 Correct 11 ms 27724 KB Output is correct
4 Correct 13 ms 27648 KB Output is correct
5 Correct 11 ms 27748 KB Output is correct
6 Correct 10 ms 27732 KB Output is correct
7 Correct 12 ms 27692 KB Output is correct
8 Correct 12 ms 27652 KB Output is correct
9 Correct 13 ms 27648 KB Output is correct
10 Correct 12 ms 27848 KB Output is correct
11 Correct 11 ms 27764 KB Output is correct
12 Correct 11 ms 27660 KB Output is correct
13 Correct 264 ms 27772 KB Output is correct
14 Correct 281 ms 27772 KB Output is correct
15 Correct 205 ms 27732 KB Output is correct
16 Correct 203 ms 27776 KB Output is correct
17 Correct 197 ms 27664 KB Output is correct
18 Correct 305 ms 27772 KB Output is correct
19 Correct 278 ms 27772 KB Output is correct
20 Correct 210 ms 27768 KB Output is correct
21 Correct 263 ms 27768 KB Output is correct
22 Correct 261 ms 27780 KB Output is correct

Subtask #3
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 68 ms 45420 KB Output is correct
3 Correct 68 ms 45416 KB Output is correct
4 Correct 45 ms 33472 KB Output is correct
5 Correct 69 ms 44364 KB Output is correct
6 Correct 70 ms 44364 KB Output is correct
7 Correct 47 ms 33540 KB Output is correct
8 Correct 47 ms 33500 KB Output is correct
9 Correct 45 ms 33500 KB Output is correct

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 43 ms 33520 KB Output is correct
3 Correct 59 ms 38176 KB Output is correct
4 Correct 44 ms 33504 KB Output is correct
5 Correct 44 ms 33548 KB Output is correct
6 Correct 66 ms 39820 KB Output is correct

Subtask #5
0.0 / 13.0

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

Subtask #6
0.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 11 ms 27732 KB Output is correct
3 Correct 11 ms 27724 KB Output is correct
4 Correct 13 ms 27648 KB Output is correct
5 Correct 11 ms 27748 KB Output is correct
6 Correct 10 ms 27732 KB Output is correct
7 Correct 12 ms 27692 KB Output is correct
8 Correct 12 ms 27652 KB Output is correct
9 Correct 13 ms 27648 KB Output is correct
10 Correct 12 ms 27848 KB Output is correct
11 Correct 11 ms 27764 KB Output is correct
12 Correct 11 ms 27660 KB Output is correct
13 Correct 264 ms 27772 KB Output is correct
14 Correct 281 ms 27772 KB Output is correct
15 Correct 205 ms 27732 KB Output is correct
16 Correct 203 ms 27776 KB Output is correct
17 Correct 197 ms 27664 KB Output is correct
18 Correct 305 ms 27772 KB Output is correct
19 Correct 278 ms 27772 KB Output is correct
20 Correct 210 ms 27768 KB Output is correct
21 Correct 263 ms 27768 KB Output is correct
22 Correct 261 ms 27780 KB Output is correct
23 Incorrect 23 ms 30676 KB Output isn't correct
24 Halted 0 ms 0 KB -

Subtask #7
0.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 11 ms 27732 KB Output is correct
2 Correct 11 ms 27732 KB Output is correct
3 Correct 11 ms 27724 KB Output is correct
4 Correct 13 ms 27648 KB Output is correct
5 Correct 11 ms 27748 KB Output is correct
6 Correct 10 ms 27732 KB Output is correct
7 Correct 12 ms 27692 KB Output is correct
8 Correct 12 ms 27652 KB Output is correct
9 Correct 13 ms 27648 KB Output is correct
10 Correct 12 ms 27848 KB Output is correct
11 Correct 11 ms 27764 KB Output is correct
12 Correct 11 ms 27660 KB Output is correct
13 Correct 264 ms 27772 KB Output is correct
14 Correct 281 ms 27772 KB Output is correct
15 Correct 205 ms 27732 KB Output is correct
16 Correct 203 ms 27776 KB Output is correct
17 Correct 197 ms 27664 KB Output is correct
18 Correct 305 ms 27772 KB Output is correct
19 Correct 278 ms 27772 KB Output is correct
20 Correct 210 ms 27768 KB Output is correct
21 Correct 263 ms 27768 KB Output is correct
22 Correct 261 ms 27780 KB Output is correct
23 Correct 68 ms 45420 KB Output is correct
24 Correct 68 ms 45416 KB Output is correct
25 Correct 45 ms 33472 KB Output is correct
26 Correct 69 ms 44364 KB Output is correct
27 Correct 70 ms 44364 KB Output is correct
28 Correct 47 ms 33540 KB Output is correct
29 Correct 47 ms 33500 KB Output is correct
30 Correct 45 ms 33500 KB Output is correct
31 Correct 43 ms 33520 KB Output is correct
32 Correct 59 ms 38176 KB Output is correct
33 Correct 44 ms 33504 KB Output is correct
34 Correct 44 ms 33548 KB Output is correct
35 Correct 66 ms 39820 KB Output is correct
36 Incorrect 54 ms 41444 KB Output isn't correct
37 Halted 0 ms 0 KB -