Submission #590426

#	Time	Username	Problem	Language	Result	Execution time	Memory
590426		Lobo	The short shank; Redemption (BOI21_prison)	C++17	15 / 100	1 ms	468 KiB

prison

#include<bits/stdc++.h>
using namespace std;

const long long inf = (long long) 1e18 + 10;
const int inf1 = (int) 1e9 + 10;
#define int long long
#define dbl long double
#define endl '\n'
#define sc second
#define fr first
#define mp make_pair
#define pb push_back
#define all(x) x.begin(), x.end()

const int maxn = 550;

int n, d, T, t[maxn], r[maxn], qtd[maxn], trT[4*maxn], lzqtd[4*maxn];
set<pair<int,int>> st;
pair<int,int> trqtd[4*maxn];

void buildT(int no, int l, int r) {
    trT[no] = inf;
    if(l != r) {
        int f1=2*no;
        int f2=2*no+1;
        int mid = (l+r)>>1;
        buildT(f1,l,mid);
        buildT(f2,mid+1,r);
    }
}

void attT(int no, int l, int r, int pos, int val) {
    if(l > pos || r < pos) return;
    if(l == r) {
        trT[no] = val;
        return;
    }
    int f1=2*no;
    int f2=2*no+1;
    int mid = (l+r)>>1;
    attT(f1,l,mid,pos,val);
    attT(f2,mid+1,r,pos,val);
    trT[no] = min(trT[f1],trT[f2]);
}

int findT(int no, int l, int r, int val) {
    if(l == r) return l;
    int f1 = 2*no;
    int f2 = 2*no+1;
    int mid = (l+r)/2;
    if(trT[f2] <= val) return findT(f2,mid+1,r,val);
    return findT(f1,l,mid,val);
}

void buildqtd(int no, int l, int r) {
    if(l == r) {
        trqtd[no] = mp(0,l);
        return;
    }
    int f1=2*no;
    int f2=2*no+1;
    int mid = (l+r)>>1;
    buildqtd(f1,l,mid);
    buildqtd(f2,mid+1,r);
    trqtd[no] = max(trqtd[f1],trqtd[f2]);
}

void flushqtd(int no, int l, int r) {
    trqtd[no].fr+= lzqtd[no];
    if(l != r) {
        int f1=2*no;
        int f2=2*no+1;
        int mid = (l+r)>>1;
        lzqtd[f1]+=lzqtd[no];
        lzqtd[f2]+=lzqtd[no];
    }
    lzqtd[no] = 0;
}

//quero pegar o cara com maior qtd em cada momento
void attqtd(int no, int l, int r, int L, int R, int val) {
    flushqtd(no,l,r);
    if(l > R || r < L) return;
    if(l >= L && r <= R) {
        lzqtd[no] = val;
        flushqtd(no,l,r);
        return;
    }
    int f1=2*no;
    int f2=2*no+1;
    int mid = (l+r)>>1;
    attqtd(f1,l,mid,L,R,val);
    attqtd(f2,mid+1,r,L,R,val);
    trqtd[no] = max(trqtd[f1],trqtd[f2]);
}

void solve() {
    cin >> n >> d >> T;
    d = min(d,n-1);
    int ans = 0;
    for(int i = 1; i <= n; i++) cin >> t[i];
    buildT(1,1,n);
    buildqtd(1,1,n);
    for(int i = 1; i <= n; i++) {
        attT(1,1,n,i,t[i]-i);
        if(trT[1] > T-i) {
            r[i] = n+1;
        }
        else {
            r[i] = findT(1,1,n,T-i);
        }

        if(r[i]!=n+1) ans++;

        //i aumenta o qtd de r[i] até i-1
        if(r[i] != i && r[i] != n+1) {
            attqtd(1,1,n,r[i],i-1,+1);
        }

        // cout << i << " " << r[i] << endl;
    }

    //qtd[i] -> qtd de retas passando de i+1 pra i
    for(int i = 1; i < n; i++) {
        qtd[i] = 0;
        for(int j = i+1; j <= n; j++) {
            if(r[j] != 0 && r[j] <= i) qtd[i]++;
        }
        st.insert(mp(qtd[i],i));
        // cout << i << " - " << qtd[i] << endl;
    }
    while(d--) {
        // auto it = st.end(); it--;
        // int id = it->sc;
        // st.erase(it);
        flushqtd(1,1,n);
        int id = trqtd[1].sc;
        //tira todos os j que tem r[j] <= id
        for(int j = id+1; j <= n; j++) {
            if(r[j] != n+1 && r[j] <= id) {
                //diminui a qtd dos que ele tinha aumentado
                // for(int i = r[j]; i < j; i++) {
                //     st.erase(mp(qtd[i],i));
                //     qtd[i]--;
                //     st.insert(mp(qtd[i],i));
                // }

                if(r[j] != j) {
                    attqtd(1,1,n,r[j],j-1,-1);
                }
                r[j] = n+1;
                ans--;
            }
        }
    }
    cout << ans << endl;
}

int32_t main() {
    ios::sync_with_stdio(false); cin.tie(0);

    // freopen("in.in", "r", stdin);
    // freopen("out.out", "w", stdout);
    int tt = 1;
    // cin >> tt;
    while(tt--) {
        solve();
    }

}

Compilation message (stderr)

prison.cpp: In function 'void flushqtd(long long int, long long int, long long int)':
prison.cpp:73:13: warning: unused variable 'mid' [-Wunused-variable]
   73 |         int mid = (l+r)>>1;
      |             ^~~

• Subtask 1: 15 / 15
• Subtask 2: 0 / 10
• Subtask 3: 0 / 20
• Subtask 4: 0 / 10
• Subtask 5: 0 / 25
• Subtask 6: 0 / 20