#include <iostream>
#include <vector>
#include <set>
#include <unordered_set>
#include <algorithm>
//#define debug(x) cerr << #x << " " << x << "\n"
using namespace std;
typedef long long ll;
typedef pair <int, int> pii;
const int bSize = 11;
const int bits = 17;
const int NMAX = 100000;
const int VMAX = NMAX * 89;
int a[NMAX], n;
int u;
int f[NMAX];
int changer = 0;
int par[NMAX];
struct BLOCK {
int time;
vector <int> v;
};
vector <BLOCK> blocks[NMAX];
vector <pii> changes[NMAX];
vector <int> vecini[NMAX];
void baga(int pt, int x, int cand) {
vecini[pt].push_back(x);
changes[pt].push_back({x, cand});
if(changes[pt].size() % bSize == 0) {
BLOCK nou;
changer++;
nou.v.clear();
nou.time = cand;
for(auto x : vecini[pt]) {
par[x] = 0;
}
for(auto x : vecini[pt]) {
par[x]++;
}
for(auto x : vecini[pt]) {
if(par[x] % 2 == 1 && f[x] != changer) {
nou.v.push_back(x);
f[x] = changer;
}
}
vecini[pt] = nou.v; // asta e confirmed?
blocks[pt].push_back(nou);
}
}
void init(int N, int D, int H[]) {
n = N;
for(int i = 0; i < N; i++) {
a[i] = H[i];
}
}
void curseChanges(int U, int A[], int B[]) {
u = U;
for(int i = 0; i < U; i++) {
int timp = i + 1;
baga(A[i], B[i], timp);
baga(B[i], A[i], timp);
}
}
bool cmp(int A, int B) {
if(a[A] != a[B])
return a[A] < a[B];
return A < B;
}
int impX[NMAX];
int impY[NMAX];
int intrebare;
vector <int> parcurgere[2];
int question(int x, int y, int v) {
int r = -1;
parcurgere[0].clear();
parcurgere[1].clear();
intrebare++;
int timpX = 0, timpY = 0;
int pas = (1 << 13), r1 = -1;
while(pas) {
if(r + pas < blocks[x].size() &&blocks[x][r + pas].time <= v)
r += pas;
if(r1 + pas < blocks[y].size() && blocks[y][r1 + pas].time <= v)
r1 += pas;
pas /= 2;
}
if(r != -1) {
timpX =blocks[x][r].time + 1;
parcurgere[0] = blocks[x][r].v;
}
if(r1 != -1) {
timpY = blocks[y][r1].time + 1;
parcurgere[1] = blocks[y][r1].v;
}
r = -1, pas = (1 << bits), r1 = -1;
while(pas) {
if(r + pas < changes[x].size() && changes[x][r + pas].second < timpX)
r += pas;
if(r1 + pas < changes[y].size() && changes[y][r1 + pas].second < timpY)
r1 += pas;
pas /= 2;
}
r++;
for(; r < changes[x].size() && changes[x][r].second <= v; r++) {
pii change = changes[x][r];
parcurgere[0].push_back(change.first);
}
r1++;
for(; r1 < changes[y].size() && changes[y][r1].second <= v; r1++) {
pii change = changes[y][r1];
parcurgere[1].push_back(change.first);
}
if(!parcurgere[0].size() || !parcurgere[1].size()) return 1e9;
sort(parcurgere[0].begin(), parcurgere[0].end(), cmp);
sort(parcurgere[1].begin(), parcurgere[1].end(), cmp);
for(auto x : parcurgere[0]) {
impX[x] = 0;
}
for(auto x : parcurgere[0]) {
impX[x]++;
}
for(auto x : parcurgere[1]) {
impY[x] = 0;
}
for(auto x : parcurgere[1]) {
impY[x]++;
}
int minim = 1e9;
int j = 0, i = 0;
while(i < parcurgere[0].size() && j < parcurgere[1].size()) {
if(impX[parcurgere[0][i]] % 2 == 0) {
i++;
continue;
}
if(impY[parcurgere[1][j]] % 2 == 0) {
j++;
continue;
}
minim = min(minim, abs(a[parcurgere[0][i]] - a[parcurgere[1][j]]));
if(a[parcurgere[0][i]] < a[parcurgere[1][j]])
i++;
else
j++;
}
return minim;
}
int main() {
int N, D, U, Q;
std::ios::sync_with_stdio(false);
std::cin.tie(NULL);
std::cin >> N >> D >> U >> Q;
int *F = new int[N];
for (int i=0; i<N; i++)
std::cin >> F[i];
init(N, D, F);
int *A = new int[U], *B = new int[U];
for (int i=0; i<U; i++) {
std::cin >> A[i] >> B[i];
}
curseChanges(U, A, B);
bool correct = true;
for (int i=0; i<Q; i++) {
int X,Y,V,CorrectAnswer;
std::cin >> X >> Y >> V;
cout << question(X,Y,V) << endl;
}
return 0;
}
Compilation message
potion.cpp: In function 'int question(int, int, int)':
potion.cpp:92:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<BLOCK>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
92 | if(r + pas < blocks[x].size() &&blocks[x][r + pas].time <= v)
| ~~~~~~~~^~~~~~~~~~~~~~~~~~
potion.cpp:94:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<BLOCK>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
94 | if(r1 + pas < blocks[y].size() && blocks[y][r1 + pas].time <= v)
| ~~~~~~~~~^~~~~~~~~~~~~~~~~~
potion.cpp:108:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
108 | if(r + pas < changes[x].size() && changes[x][r + pas].second < timpX)
| ~~~~~~~~^~~~~~~~~~~~~~~~~~~
potion.cpp:110:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
110 | if(r1 + pas < changes[y].size() && changes[y][r1 + pas].second < timpY)
| ~~~~~~~~~^~~~~~~~~~~~~~~~~~~
potion.cpp:115:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
115 | for(; r < changes[x].size() && changes[x][r].second <= v; r++) {
| ~~^~~~~~~~~~~~~~~~~~~
potion.cpp:120:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
120 | for(; r1 < changes[y].size() && changes[y][r1].second <= v; r1++) {
| ~~~^~~~~~~~~~~~~~~~~~~
potion.cpp:141:13: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
141 | while(i < parcurgere[0].size() && j < parcurgere[1].size()) {
| ~~^~~~~~~~~~~~~~~~~~~~~~
potion.cpp:141:41: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
141 | while(i < parcurgere[0].size() && j < parcurgere[1].size()) {
| ~~^~~~~~~~~~~~~~~~~~~~~~
potion.cpp: In function 'int main()':
potion.cpp:178:19: warning: unused variable 'CorrectAnswer' [-Wunused-variable]
178 | int X,Y,V,CorrectAnswer;
| ^~~~~~~~~~~~~
potion.cpp:176:10: warning: unused variable 'correct' [-Wunused-variable]
176 | bool correct = true;
| ^~~~~~~
/usr/bin/ld: /tmp/ccuwGcOT.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccWidz4S.o:potion.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
