Submission #794059

# Submission time Handle Problem Language Result Execution time Memory
794059 2023-07-26T09:00:27 Z JakobZorz The Potion of Great Power (CEOI20_potion) C++14
38 / 100
1169 ms 262144 KB
#include<iostream>
#include<set>
#include<vector>
#include<unordered_set>
using namespace std;
typedef long long ll;

const int C=64;

// time travelling multiset
class TTMS{
    vector<int>days;
    vector<pair<int,int>>changes;
    vector<multiset<int>>checkpoints;
    multiset<int>curr_set;
    int curr_day=0;
    void add_day(){
        days.push_back(curr_day);
        while(checkpoints.size()*C<curr_day)
            checkpoints.push_back(curr_set);
    }
public:
    TTMS(){
        checkpoints={{}};
    }
    
    void add_new_day(int day){
        curr_day=day;
    }
    
    void insert(int el){
        add_day();
        curr_set.insert(el);
        changes.push_back({el,1});
    }
    
    void remove(int el){
        add_day();
        curr_set.erase(curr_set.find(el));
        changes.push_back({el,-1});
    }
    
    vector<int> get(int day){

        multiset<int>s;
        if(day/C>=checkpoints.size())
            s=curr_set;
        else
            s=checkpoints[day/C];
        for(int i=0;i<(int)changes.size();i++){
            if(days[i]>day)
                break;
            if(days[i]>day/C*C){
                if(changes[i].second==1)
                    s.insert(changes[i].first);
                else if(changes[i].second==-1)
                    s.erase(s.find(changes[i].first));
            }
        }
        return{s.begin(),s.end()};
    }
};

int get_closest(vector<int>a,vector<int>b){
    int res=1e9;
    int i=0,j=0;
    while(i<(int)a.size()&&j<(int)b.size()){
        res=min(res,abs(a[i]-b[j]));
        if(i<(int)a.size()&&a[i]<=b[j])
            i++;
        else if(j<(int)b.size()&&b[j]<=a[i])
            j++;
    }
    return res;
}

int n;
int heights[100000];
TTMS ttms[100000];

void init(int N, int D, int H[]) {
    n=N;
    for(int i=0;i<n;i++)
        heights[i]=H[i];
}

ll to(int a,int b){
    if(a<b)
        swap(a,b);
    return a+((ll)b<<32);
}

void curseChanges(int U, int A[], int B[]) {
    unordered_set<ll>conns;
    for(int day=0;day<U;day++){
        int a=A[day],b=B[day];
        ll conn=to(a,b);
        ttms[a].add_new_day(day+1);
        ttms[b].add_new_day(day+1);
        if(conns.find(conn)==conns.end()){
            conns.insert(conn);
            ttms[a].insert(heights[b]);
            ttms[b].insert(heights[a]);
        }else{
            conns.erase(conn);
            ttms[a].remove(heights[b]);
            ttms[b].remove(heights[a]);
        }
    }
}

int question(int x, int y, int v) {
    vector<int>a=ttms[x].get(v);
    vector<int>b=ttms[y].get(v);
    
    return get_closest(a,b);
}

Compilation message

potion.cpp: In member function 'void TTMS::add_day()':
potion.cpp:19:35: warning: comparison of integer expressions of different signedness: 'std::vector<std::multiset<int> >::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   19 |         while(checkpoints.size()*C<curr_day)
      |               ~~~~~~~~~~~~~~~~~~~~^~~~~~~~~
potion.cpp: In member function 'std::vector<int> TTMS::get(int)':
potion.cpp:46:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::multiset<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |         if(day/C>=checkpoints.size())
      |            ~~~~~^~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 11 ms 19024 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 13 ms 20048 KB Output is correct
2 Correct 14 ms 20176 KB Output is correct
3 Correct 13 ms 20112 KB Output is correct
4 Correct 22 ms 20960 KB Output is correct
# Verdict Execution time Memory Grader output
1 Runtime error 185 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 184 ms 262144 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 158 ms 112872 KB Output is correct
2 Correct 122 ms 29500 KB Output is correct
3 Correct 194 ms 24656 KB Output is correct
4 Correct 778 ms 87912 KB Output is correct
5 Correct 1169 ms 102556 KB Output is correct
6 Correct 248 ms 96456 KB Output is correct
7 Correct 676 ms 40208 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 11 ms 19024 KB Output is correct
2 Correct 13 ms 20048 KB Output is correct
3 Correct 14 ms 20176 KB Output is correct
4 Correct 13 ms 20112 KB Output is correct
5 Correct 22 ms 20960 KB Output is correct
6 Runtime error 185 ms 262144 KB Execution killed with signal 9
7 Halted 0 ms 0 KB -