Submission #1013002

# Submission time Handle Problem Language Result Execution time Memory
1013002 2024-07-03T04:42:19 Z octane Meteors (POI11_met) C++17
74 / 100
2649 ms 62776 KB
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
struct qu{
    int l, r;
    ll a;
    qu(){}
    qu(int l, int r, ll a): l(l), r(r), a(a){}
};
typedef pair<int, int> pi;
typedef pair<pi, vector<int>> pii;

int n, k, q, x;
ll goal[300005];
qu qus[300005];
ll tree[1200005];
vector<int> num[300005];
int res[300005];
vector<pii> vec[50];

void init(int node, int s, int e){
    tree[node] = 0;
    if(s==e) return;
    int m = (s+e)>>1;
    init(node<<1, s, m);
    init(node<<1|1, m+1, e);
}

void update(int node, int s, int e, int l, ll v){
    if(e<l||l<s) return;
    if(s==e){
        tree[node] += v;
        return;
    }
    int m = (s+e)>>1;
    update(node<<1, s, m, l, v);
    update(node<<1|1, m+1, e, l, v);
    tree[node] = tree[node<<1]+tree[node<<1|1];
}

ll query(int node, int s, int e, int l, int r){
    if(e<l||r<s) return 0;
    if(l<=s&&e<=r) return tree[node];
    int m = (s+e)>>1;
    return query(node<<1, s, m, l, r)+query(node<<1|1, m+1, e, l, r);
}

int main(){
    scanf("%d %d", &n, &k);
    for(int i=1;i<=k;i++){
        scanf("%d", &x);
        num[x].push_back(i);
    }
    for(int i=1;i<=n;i++) scanf("%lld", &goal[i]);
    scanf("%d", &q);
    vector<int> emp;
    for(int i=1;i<=n;i++) emp.push_back(i);
    vec[0].push_back(pii(pi(0, q-1), emp));
    for(int i=0;i<q;i++) scanf("%d %d %lld", &qus[i].l, &qus[i].r, &qus[i].a);
    int idx = 0;
    while(!vec[idx].empty()){
        for(int i=1;i<=4*k;i++) tree[i] = 0;
        int prv = 0;
        for(auto p: vec[idx]){
            int s = p.first.first;
            int e = p.first.second;
            int m = (s+e)>>1;
            vector<int> v = p.second;
            if(s==e){
                for(int i=prv;i<=s;i++){
                    if(qus[i].l<=qus[i].r){
                        update(1, 1, k+1, qus[i].l, qus[i].a);
                        update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                    }
                    else{
                        update(1, 1, k+1, qus[i].l, qus[i].a);
                        update(1, 1, k+1, k+1, -qus[i].a);
                        update(1, 1, k+1, 1, qus[i].a);
                        update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                    }
                }
                for(auto u: v){
                    ll sm = 0;
                    for(auto w: num[u]) sm += query(1, 1, k+1, 1, w);
                    if(sm>=goal[u]) res[u] = s;
                    else res[u] = -1;
                }
                prv = e+1;
                continue;
            }
            vector<int> vec1;
            vector<int> vec2;
            for(int i=prv;i<=m;i++){
                if(qus[i].l<=qus[i].r){
                    update(1, 1, k+1, qus[i].l, qus[i].a);
                    update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                }
                else{
                    update(1, 1, k+1, qus[i].l, qus[i].a);
                    update(1, 1, k+1, k+1, -qus[i].a);
                    update(1, 1, k+1, 1, qus[i].a);
                    update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                }
            }
            for(auto u: v){
                ll sm = 0;
                for(auto w: num[u]) sm += query(1, 1, k+1, 1, w);
                if(sm>=goal[u]) vec1.push_back(u);
                else vec2.push_back(u);
            }
            for(int i=m+1;i<=e;i++){
                if(qus[i].l<=qus[i].r){
                    update(1, 1, k+1, qus[i].l, qus[i].a);
                    update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                }
                else{
                    update(1, 1, k+1, qus[i].l, qus[i].a);
                    update(1, 1, k+1, k+1, -qus[i].a);
                    update(1, 1, k+1, 1, qus[i].a);
                    update(1, 1, k+1, qus[i].r+1, -qus[i].a);
                }
            }
            prv = e+1;
            vec[idx+1].push_back(pii(pi(s, m), vec1));
            vec[idx+1].push_back(pii(pi(m+1, e), vec2));
        }
        idx++;
    }
    for(int i=1;i<=n;i++){
        if(res[i]==-1) printf("NIE\n");
        else printf("%d\n", res[i]+1);
    }
}

Compilation message

met.cpp: In function 'int main()':
met.cpp:54:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   54 |     scanf("%d %d", &n, &k);
      |     ~~~~~^~~~~~~~~~~~~~~~~
met.cpp:56:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   56 |         scanf("%d", &x);
      |         ~~~~~^~~~~~~~~~
met.cpp:59:32: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   59 |     for(int i=1;i<=n;i++) scanf("%lld", &goal[i]);
      |                           ~~~~~^~~~~~~~~~~~~~~~~~
met.cpp:60:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   60 |     scanf("%d", &q);
      |     ~~~~~^~~~~~~~~~
met.cpp:64:31: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   64 |     for(int i=0;i<q;i++) scanf("%d %d %lld", &qus[i].l, &qus[i].r, &qus[i].a);
      |                          ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 4 ms 12632 KB Output is correct
2 Correct 7 ms 12700 KB Output is correct
3 Correct 4 ms 12892 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 12892 KB Output is correct
2 Correct 4 ms 12892 KB Output is correct
3 Correct 5 ms 12908 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 204 ms 20376 KB Output is correct
2 Correct 328 ms 23236 KB Output is correct
3 Correct 276 ms 22156 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 279 ms 21368 KB Output is correct
2 Correct 282 ms 21476 KB Output is correct
3 Correct 328 ms 22452 KB Output is correct
4 Correct 61 ms 17232 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 136 ms 20888 KB Output is correct
2 Correct 221 ms 23820 KB Output is correct
3 Correct 111 ms 17748 KB Output is correct
4 Correct 290 ms 23164 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 310 ms 20076 KB Output is correct
2 Correct 292 ms 21468 KB Output is correct
3 Correct 239 ms 20428 KB Output is correct
4 Correct 312 ms 24056 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 2649 ms 62776 KB Output is correct
2 Incorrect 1858 ms 52288 KB Output isn't correct
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 2608 ms 62216 KB Output is correct
2 Incorrect 1079 ms 51764 KB Output isn't correct
3 Halted 0 ms 0 KB -