oj.uz

답안 #333515

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
333515 2020-12-06T17:39:32 Z Vladth11 Meteors (POI11_met) C++14
100 / 100
 2489 ms 65268 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define debug(x) cerr << #x << " " << x << "\n"
#define debug_with_space(x) cerr << #x << " " << x << " "
 
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair <ll, ll> pii;
typedef pair <ll, pii> piii;
typedef tree <pii, null_type, less <pii>, rb_tree_tag, tree_order_statistics_node_update> OST;
 
const ll NMAX = 300005;
const ll INF = (1 << 16) - 1;
const ll MOD = 1000000007;
const ll BLOCK = 101;
const ll nr_of_bits = 35;
 
struct ura{
    int a, b, c;
};
 
vector <ura> segm;
vector <int> reg[NMAX];
vector <int> check[NMAX];
int n, m, k;
int req[NMAX], l[NMAX], r[NMAX];
 
class AIB{
    ll aib[NMAX + 5];
    void update(ll x, ll val){
        for(ll i = x; i <= m; i+=i&(-i)){
            aib[i] += val;
        }
    }
    ll query(ll x){
        ll sum = 0;
        for(ll i = x; i > 0; i-=i&(-i)){
            sum += aib[i];
        }
        return sum;
    }
public:
    void ru(ll l, ll r, ll val){
        if(l <= r){
            update(l, val);
            update(r + 1, -val);
        }
        if(l > r){
            update(l, val);
            update(1, val);
            update(r + 1, -val);
        }
    }
    ll cn(ll x){
        return query(x);
    }
    void init(){
        for(ll i = 1; i <= m; i++){
            aib[i] = 0;
        }
    }
}aib;
 
void baga(){
    aib.init();
    ll i;
    for(i = 1; i <= k; i++)
        check[i].clear();
    for(i = 1; i <= n; i++){
      	if(l[i] == r[i])
          continue;
        ll mid = (l[i] + r[i]) / 2;
        check[mid].push_back(i);
    }
    for(i = 0; i < k; i++){
        aib.ru(segm[i].a, segm[i].b, segm[i].c);
        for(auto x : check[i + 1]){
            ll s = 0;
            for(auto y : reg[x]){
                s += aib.cn(y);
              if(s > 1e9)
                break;
            }
            ll mid = i + 1;
            if(s >= req[x]){
                r[x] = mid;
            }else{
                l[x] = mid + 1;
            }
        }
    }
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    ll i, x;
    cin >> n >> m;
    for(i = 1; i <= m; i++){
        cin >> x;
        reg[x].push_back(i);
    }
    for(i = 1; i <= n; i++){
        cin >> req[i];
    }
    cin >> k;
    for(i = 1; i <= k; i++){
        ura x;
        cin >> x.a >> x.b >> x.c;
        segm.push_back(x);
    }
    for(i = 1; i <= n; i++){
        l[i] = 1;
        r[i] = k + 1;
    }
    for(ll i = 1; i <= nr_of_bits; i++){
        baga();
    }
    for(i = 1; i <= n; i++){
        if(l[i] != k + 1)
        cout << l[i] << "\n";
        else cout << "NIE\n";
    }
    return 0;
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 14444 KB Output is correct
2 Correct 12 ms 14444 KB Output is correct
3 Correct 12 ms 14444 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 14444 KB Output is correct
2 Correct 12 ms 14572 KB Output is correct
3 Correct 12 ms 14572 KB Output is correct

Subtask #3
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 118 ms 16356 KB Output is correct
2 Correct 198 ms 18148 KB Output is correct
3 Correct 172 ms 18020 KB Output is correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 161 ms 17380 KB Output is correct
2 Correct 170 ms 17504 KB Output is correct
3 Correct 205 ms 18408 KB Output is correct
4 Correct 45 ms 16620 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 138 ms 16736 KB Output is correct
2 Correct 152 ms 18792 KB Output is correct
3 Correct 95 ms 15336 KB Output is correct
4 Correct 156 ms 18276 KB Output is correct

Subtask #6
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 165 ms 15892 KB Output is correct
2 Correct 174 ms 17260 KB Output is correct
3 Correct 131 ms 16256 KB Output is correct
4 Correct 188 ms 19684 KB Output is correct

Subtask #7
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1595 ms 34212 KB Output is correct
2 Correct 1105 ms 22308 KB Output is correct
3 Correct 426 ms 17632 KB Output is correct
4 Correct 2328 ms 61480 KB Output is correct

Subtask #8
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1563 ms 33048 KB Output is correct
2 Correct 1120 ms 22120 KB Output is correct
3 Correct 358 ms 17760 KB Output is correct
4 Correct 2489 ms 65268 KB Output is correct