Source Code
// Those who cannot remember the past are
// condemned to repeat it (use DP -_-)
// - George Santayana

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define ll long long
#define ull unsigned long long
#define MOD 1000000007
#define PI 3.14159265
#define ceil(a, b) ((a / b) + (a % b ? 1 : 0))
#define imin INT_MIN
#define imax INT_MAX
#define llmax LLONG_MAX
#define llmin LLONG_MIN
#define inf 2000000000
#define nl '\n'
#define ppcnt __builtin_popcount
#define ppcntll __builtin_popcountll
#define clz __builtin_clz
#define clzll __builtin_clzll
#define ctz __builtin_ctz
#define ctzll __builtin_ctzll
#define modulo(a, b, mod) ((((a) % mod) * ((b) % mod)) % mod)
#define cnte(v, x) count(all(v), (x))
#define mine(v) min_element(all(v))
#define maxe(v) max_element(all(v))
#define debug(x) cout << "x: " << x << nl;
#define debug2(x, y) cout << "x: " << x << " y: " << y << nl;
#define ordered_set tree<int, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
#define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>

//vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
//vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};

template<typename T = int> istream& operator>>(istream& in, vector<pair<int, int>>& v){
    for (auto& [x, y] : v) in >> x >> y;
    return in;
}

template<typename T = int> istream& operator>>(istream& in, vector<T>& v){
    for (T& i : v) in >> i;
    return in;
}

template<typename T = int> ostream& operator<<(ostream& out, const vector<T>& v){
    for (const T& x : v)
        out << x << ' ';
    return out;
}

template<typename T = pair<ll, ll>> ostream& operator << (ostream& out, const vector<pair<ll, ll>>& v){
    for(auto& [x, y] : v){
        out << x << ' ' << y << nl;
    }
    return out;
}

void Start_Crushing() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
#ifndef ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif
}

void solve(){
    int n; cin >> n;
    vector<int> v(n);
    vector<pair<ll, ll>> tmp;


    for(int i = 0; i < n; i++){
        cin >> v[i];
        tmp.push_back({v[i] + (i + 1), i + 1});
    }

    sort(all(tmp), [&](pair<ll, ll>& a, pair<ll, ll>& b){
        if(a.first == b.first)
            return a.second < b.second;
        return a.first < b.first;
    });

    vector<ll> mini(tmp.size());
    mini.back() = tmp.back().second;

    for(int i = mini.size() - 2; i >= 0; i--){
        mini[i] = min(mini[i + 1], tmp[i].second);
    }

    for(int i = 0; i < n; i++){
        ll target = (i + 1) - v[i];

        ll left = -1, right = (int)tmp.size() - 1;

        while(left < right - 1){
            ll mid = left - (left - right) / 2;

            if(tmp[mid].first >= target)
                right = mid;
            else
                left = mid;
        }
        if(tmp[right].first >= target)
            cout << mini[right] << ' ';
        else
            cout << -1 << ' ';
    }
}

int main(){
    Start_Crushing();

    int t = 1;
    /*Multiple test cases?*/ cin >> t;
    while (t--) {
        solve();
        if(!t)
            break;
        cout << nl;
    }

//    for(int tc = 1; tc <= t; tc++){
//        cout << "Case #" << tc << ": ";
//        solve();
//        if(tc != t)
//            cout << nl;
//    }

    return 0;
}
Copy
Easy challenge? El_Gemmy
GNU G++17
165 ms
7.1 MB
Accepted