oj.uz

Submission #920496

# Submission time Handle Problem Language Result Execution time Memory
920496 2024-02-02T15:44:55 Z NotLinux Meteors (POI11_met) C++17
87 / 100
 1871 ms 65536 KB 
#include <bits/stdc++.h>
using namespace std;
#define lng __int128
const int gh = 21;
struct FENWICK{
    vector < lng > fen;
    int sz;
    FENWICK(int x){
        sz = x+3;
        fen.assign(sz,0);
    }
    inline void update(int p , int x)  {
        while(p <= sz){
            fen[p] += x;
            p += p & (-p);
            //cout << "update:" << p << " " << sz << endl;
        }
    }
    inline lng query(int x){
        lng ret = 0;
        while(x != 0){
            ret += fen[x];
            x -= x & (-x);
            //cout <<"query:"<< x << endl;
        }
        return ret;
    }
    inline void rangeupdate(int l , int r,int x){
        update(l,x);
        if(r != sz)update(r+1,-x);
    }
};
 
void solve(){
    int n,m;cin >> n >> m;
    vector < int > devlet[n+1];
    for(int i = 1;i<=m;i++){
        int x;cin >> x;
        devlet[x].push_back(i);
    }
    int need[n+1];
    for(int i = 1;i<=n;i++){
        cin >> need[i];
    }
    int q;cin >> q;
    vector < array < int , 3 > > query(q+1);
    for(int i = 1;i<=q;i++){
        cin >> query[i][0] >> query[i][1] >> query[i][2];
    }
 
    vector < int > paralel[q+2],nparalel[q+2];
    int l[n+1] , r[n+1];
    for(int i = 1;i<=n;i++){
        l[i] = 1;
        r[i] = q+1;
        paralel[(l[i] + r[i])/2].push_back(i);
    }
    for(int i = 0;i<=gh;i++){
        FENWICK fen(m+10);
        for(int j = 1;j<=q;j++){
            if(query[j][0] <= query[j][1]){
                fen.rangeupdate(query[j][0],query[j][1],query[j][2]);
            }
            else{
                fen.rangeupdate(1,query[j][1],query[j][2]);
                fen.rangeupdate(query[j][0],m+1,query[j][2]);
            }
            for(auto itr : paralel[j]){
                if(l[itr] == r[itr]){
                    nparalel[j].push_back(itr);
                    continue;
                }
                lng cur = 0;
                for(auto itr1 : devlet[itr]){
                    cur += fen.query(itr1);
                }
                int mid = (l[itr] + r[itr])/2;
                if(cur >= need[itr]){//r = mid;
                    r[itr] = mid;
                }
                else {//l = mid
                    l[itr] = mid+1;
                }
                nparalel[(l[itr] + r[itr]) / 2].push_back(itr);
            }
        }
        for(int j = 1;j<=q;j++){
            paralel[j].swap(nparalel[j]);
            nparalel[j].clear();
        }
    }
    int ans[n+1];
    memset(ans , -1 , sizeof(ans));
    for(int j = 1;j<=q;j++){
        for(auto itr : paralel[j]){
            ans[itr] = j;
        }
    }
    for(int i = 1;i<=n;i++){
        if(ans[i] == -1)cout << "NIE" << '\n';
        else cout << ans[i] << '\n';
    }
}
signed main(){
    ios_base::sync_with_stdio(0);cin.tie(0);
    int testcase = 1;//cin >> testcase;
    while(testcase--)solve();
}

Subtask #1
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 2 ms 348 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 2 ms 604 KB Output is correct

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 71 ms 5052 KB Output is correct
2 Correct 118 ms 7612 KB Output is correct
3 Correct 108 ms 7112 KB Output is correct

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 112 ms 6468 KB Output is correct
2 Correct 106 ms 6416 KB Output is correct
3 Correct 123 ms 7680 KB Output is correct
4 Correct 26 ms 3656 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 76 ms 5440 KB Output is correct
2 Correct 90 ms 8260 KB Output is correct
3 Correct 55 ms 3448 KB Output is correct
4 Correct 101 ms 7596 KB Output is correct

Subtask #6
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 107 ms 4444 KB Output is correct
2 Correct 108 ms 6424 KB Output is correct
3 Correct 83 ms 4884 KB Output is correct
4 Correct 128 ms 9272 KB Output is correct

Subtask #7
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 975 ms 39352 KB Output is correct
2 Correct 546 ms 23992 KB Output is correct
3 Correct 297 ms 15484 KB Output is correct
4 Correct 1843 ms 65536 KB Output is correct

Subtask #8
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 960 ms 37648 KB Output is correct
2 Correct 530 ms 24048 KB Output is correct
3 Correct 228 ms 12052 KB Output is correct
4 Runtime error 1871 ms 65536 KB Execution killed with signal 9
5 Halted 0 ms 0 KB -