1494C - 1D Sokoban - CodeForces Solution

#include <bits/stdc++.h>
using namespace std;
int calc(const vector<int> &a, const vector<int> &b){
    int n = a.size();
    int m = b.size();
    vector<int> su(n + 1);
    int r = m - 1;
    for (int i = n - 1; i >= 0; --i){
        su[i] = su[i + 1];
        while (r >= 0 && b[r] > a[i]) --r;
        if (r >= 0 && b[r] == a[i]) ++su[i];
    }
    int ans = 0;
    int j = 0;
    r = 0;
    for(int l=0;l< m;l++){
        while (j < n && a[j] <= b[l] + j) ++j;
        while (r < m && b[r] - b[l] < j) ++r;
        ans = max(ans, r - l + su[j]);
    }
    return ans;
}

int main() {
    int t;
    cin>>t;
    while( t--){
        int n, m;
        cin>>n>>m;
        vector<int> a(n), b(m);
        for(int i=0;i< n;i++) cin>>a[i];
        for(int i=0;i< m;i++) cin>>b[i];
        vector<int> al, bl, ar, br;
        for(int i=0;i< n;i++){
            if (a[i] < 0) al.push_back(-a[i]);//lo ve como si estuvieran
            //a la derecha las cajas de la izquierda
            else ar.push_back(a[i]);//posición de las cajas de la derecha
        }
        for(int i=0;i< m;i++){
            if (b[i] < 0) bl.push_back(-b[i]);//lo ve como si estuvieran
            //a la derecha las posiciones especiales
            else br.push_back(b[i]);//posiciones especiales de la derecha
        }
        sort(al.begin(), al.end());//lo ordena de menor a mayor
        sort(bl.begin(), bl.end());//lo ordena de menor a mayor
        cout<<calc(al, bl) + calc(ar, br)<<"\n";
    }
    return 0;
}