oj.uz

Submission #491707

# Submission time Handle Problem Language Result Execution time Memory
491707 2021-12-03T22:42:47 Z faresbasbs Meteors (POI11_met) C++14
36 / 100
 6000 ms 37292 KB 
#include <bits/stdc++.h>
using namespace std;

struct p{
    long long l1,r1,l2,r2,val;
};

const int sq = 300001;
long long n,m,k,t,type[300001],val[300001],sum[300001];
pair<long long,long long> lst[300001];
vector<long long> segment,all[300001];
bool done[300001];
p event[300001];

void upd(int a , int b , int v , int curr , int l , int r){
    if(b < l || a > r){
        return ;
    }
    if(a <= l && b >= r){
        segment[curr] += v;
        return ;
    }
    int mid = (l+r)/2;
    upd(a,b,v,2*curr,l,mid),upd(a,b,v,2*curr+1,mid+1,r);
}

long long getsum(int idx){
    idx += t-1;
    long long ret = 0;
    while(idx){
        ret += segment[idx];
        idx /= 2;
    }
    return ret;
}

long long num(int l , int r , int idx){
    return (upper_bound(all[idx].begin(),all[idx].end(),r)-all[idx].begin())-(lower_bound(all[idx].begin(),all[idx].end(),l)-all[idx].begin());
}

long long getval(int idx , int ev){
    long long numb = num(event[ev].l1,event[ev].r1,idx)+(event[ev].l2 == -1 ? 0ll : num(event[ev].l2,event[ev].r2,idx));
    return event[ev].val*numb;
}

void check(int idx){
    for(int i = 1 ; i <= n ; i += 1){
        sum[i] = 0;
    }
    for(int i = 1 ; i <= m ; i += 1){
        if(done[type[i]]){
            continue;
        }
        sum[type[i]] += getsum(i);
    }
    for(int i = 1 ; i <= n ; i += 1){
        if(sum[i] >= val[i]){
            done[i] = 1;
        }
        if(done[i]){
            continue;
        }
        lst[i] = {sum[i],idx};
    }
}

int main(){
    ios_base::sync_with_stdio(false);
    cout.tie(NULL);
    cin.tie(NULL);
    cin >> n >> m;
    t = pow(2,ceil(log2(m)));
    segment.resize(2*t,0);
    for(int i = 1 ; i <= m ; i += 1){
        cin >> type[i];
        all[type[i]].push_back(i);
    }
    for(int i = 1 ; i <= n ; i += 1){
        cin >> val[i];
        lst[i] = {0,0};
    }
    cin >> k;
    for(int i = 1 ; i <= k ; i += 1){
        int l,r,v;
        cin >> l >> r >> v;
        if(r >= l){
            event[i] = {l,r,-1,-1,v};
            upd(l,r,v,1,1,t);
        }else{
            event[i] = {l,m,1,r,v};
            upd(l,m,v,1,1,t),upd(1,r,v,1,1,t);
        }
        if(i%sq == 0){
            check(i);
        }
    }
    //check(n);
    for(int i = 1 ; i <= n ; i += 1){
        pair<long long,long long> curr = lst[i];
        bool ok = 1;
        while(curr.first < val[i]){
            curr.second += 1;
            if(curr.second == m+1){
                ok = 0;
                break;
            }
            curr.first += getval(i,curr.second);
        }
        if(!ok){
            cout << "NIE\n";
        }else{
            cout << curr.second << '\n';
        }
    }
}

Subtask #1
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 7372 KB Output is correct
2 Correct 16 ms 7504 KB Output is correct
3 Correct 7 ms 7500 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 7372 KB Output is correct
2 Correct 4 ms 7460 KB Output is correct
3 Correct 13 ms 7504 KB Output is correct

Subtask #3
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1467 ms 11488 KB Output is correct
2 Execution timed out 6045 ms 12232 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #4
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 4988 ms 11712 KB Output is correct
2 Correct 5453 ms 12784 KB Output is correct
3 Correct 1606 ms 13560 KB Output is correct
4 Correct 5716 ms 10588 KB Output is correct

Subtask #5
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 1880 ms 11580 KB Output is correct
2 Execution timed out 6078 ms 13428 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #6
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 159 ms 11280 KB Output is correct
2 Correct 2672 ms 11688 KB Output is correct
3 Correct 1675 ms 11400 KB Output is correct
4 Execution timed out 6063 ms 12420 KB Time limit exceeded

Subtask #7
0 / 13.0

# Verdict Execution time Memory Grader output
1 Execution timed out 6033 ms 37292 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# Verdict Execution time Memory Grader output
1 Execution timed out 6067 ms 36792 KB Time limit exceeded
2 Halted 0 ms 0 KB -