166122025-05-06 20:58:18algoproTornyokcpp17Accepted 100/100307ms49204 KiB
// UUID: 2e5eac05-03d6-48d3-9d0d-bb1d2b7d8333
#include <bits/stdc++.h>
using namespace std;

int n, k;
vector<int> numLess, previous;
vector<vector<int>> nbrs;

int calcLess(int Indx){
    if(0<=numLess[Indx]) return numLess[Indx];
    int ans=0;
    for(int x : nbrs[Indx]) ans=max(ans, calcLess(x)+1);
    return numLess[Indx]=ans;
}

void calcLess2(){
    for (int i = n - 1; i >= 0; i--){
        if(previous[i] != -1) numLess[previous[i]] = max(numLess[previous[i]], numLess[i] + 1);
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> k;
    vector<long long> sortedArr(n+k);
    vector<int> h(n+1), towerH(k);
    for(int i=0;i<n;i++){
        cin >> h[i];
        sortedArr[i]=(((long long)(h[i])<<32)|i);
    }
    for(int i=0;i<k;i++){
        cin >> towerH[i];
        sortedArr[n+i]=(((long long)(towerH[i])<<32)|n+i);
    }
    h[n]=INT_MAX;
    sort(sortedArr.begin(), sortedArr.end());
    stack<int> stc;
    stc.push(n);
    nbrs.resize(n+1);
    previous.resize(n+1, -1);
    for(int i=0;i<n;i++){
        while(h[stc.top()]<h[i]) stc.pop();
        // nbrs[stc.top()].push_back(i); // az viszonylag lassú, hogy push_back-elsz
        previous[i] = stc.top(); // ha minden csúcsra az előzőt számolod, akkor nem kell pushbackelni
        stc.push(i);
    }
    numLess.resize(n+1, 0); // te itt eddig 0-ra resizeoltál, az egyszerűség kedvéért 0-ra állítottam a calcless2 miatt
    // calcLess(n);
    calcLess2(); // ugyan az a calc, csak a másik irányból
    int bestSoFar=0;
    vector<int>& ans=towerH;
    for(int i=0;i<n+k;i++){
        sortedArr[i]&=0xffffffff;
        if(n<=sortedArr[i]){
            towerH[sortedArr[i]-n]=bestSoFar+1;
        }
        else{
            bestSoFar=max(bestSoFar, numLess[sortedArr[i]]);
        }
    }
    for(int i=0;i<k;i++) cout << towerH[i] << ' ';
}
SubtaskSumTestVerdictTimeMemory
base100/100
1Accepted0/01ms316 KiB
2Accepted0/0241ms37428 KiB
3Accepted2/21ms508 KiB
4Accepted2/21ms316 KiB
5Accepted6/61ms316 KiB
6Accepted6/61ms316 KiB
7Accepted4/413ms2624 KiB
8Accepted4/420ms4688 KiB
9Accepted8/875ms17424 KiB
10Accepted8/8128ms27700 KiB
11Accepted5/5254ms36416 KiB
12Accepted5/5291ms43316 KiB
13Accepted5/546ms8244 KiB
14Accepted5/5101ms18740 KiB
15Accepted5/5128ms28328 KiB
16Accepted5/5206ms30772 KiB
17Accepted5/5240ms37424 KiB
18Accepted5/5287ms44596 KiB
19Accepted5/5307ms44628 KiB
20Accepted5/5266ms48708 KiB
21Accepted5/5246ms48696 KiB
22Accepted5/5289ms49204 KiB