Submission #584451

#	Time	Username	Problem	Language	Result	Execution time	Memory
584451		FatihSolak	OGLEDALA (COI15_ogledala)	C++17	19 / 100	59 ms	8504 KiB

ogledala

#include <bits/stdc++.h>
#define N 100005
using namespace std;
long long a[N],b[N];
long long len[N];
map<pair<long long,long long>,long long> rem;
long long val(long long curlen,long long target){
    if(curlen <= target)return curlen == target;
    if(rem.count({curlen,target}))
        return rem[{curlen,target}];
    return rem[{curlen,target}] = val((curlen-1)/2,target) + val(curlen/2,target);
    
}
long long get(long long st,long long curlen,long long target,long long idx){
    if(curlen == target)return st + (target - 1)/2;
    //cout << st << " " << curlen << " " << target << " " << idx << endl;
    long long nwlen = (curlen - 1)/2;
    if(val(nwlen,target) >= idx){
        return get(st,nwlen,target,idx);
    }
    idx -= val(nwlen,target);
    nwlen = (curlen)/2;
    return get(st + (curlen + 1)/2,nwlen,target,idx);
}
void solve(){
    int m,n,q;
    cin >> m >> n >> q;
    map<long long,vector<pair<long long,long long>>> mp; 
    set<long long> points;
    for(int i = 1;i<=n;i++){
        cin >> a[i];
    }
    a[n+1] = m + 1;
    for(int i = 0;i<=n;i++){
        len[i] = a[i+1] - a[i] - 1;
        map<long long,long long> cnt;
        cnt[len[i]] = 1;
        set<long long> nums = {len[i]};
        while(*nums.rbegin() > 0){
            long long nowlen = *nums.rbegin();
            nums.erase(nowlen);
            points.insert(nowlen);
            mp[nowlen].push_back({i,cnt[nowlen]});
            nums.insert((nowlen-1)/2);
            nums.insert(nowlen/2);
            cnt[(nowlen-1)/2] += cnt[nowlen];
            cnt[nowlen/2] += cnt[nowlen];
        }
    }
    for(auto &u:mp){
        reverse(u.second.begin(),u.second.end());
    }
    long long nowgo = n;
    for(int i = 1;i<=q;i++){
        cin >> b[i];
        if(b[i] <= n){
            cout << a[b[i]] << "\n";
            continue;
        }
        while(nowgo + mp[*points.rbegin()].back().second < b[i]){
            nowgo += mp[*points.rbegin()].back().second;
            mp[*points.rbegin()].pop_back();
            if(mp[*points.rbegin()].empty()){
                points.erase(*points.rbegin());
            }
        }   
        b[i] -= nowgo;
        cout << get( a[mp[*points.rbegin()].back().first] + 1,len[mp[*points.rbegin()].back().first],*points.rbegin(),b[i]) << "\n";
    }

}
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    #ifdef Local
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int t = 1;
    //cin >> t;
    while(t--){
        solve();
    }
    #ifdef Local
        cout << endl << fixed << setprecision(2) << 1000.0*clock()/CLOCKS_PER_SEC << " milliseconds.";
    #endif
}

• Subtask 1: 19 / 19
• Subtask 2: 0 / 22
• Subtask 3: 0 / 59