oj.uz

Submission #1103499

# Submission time Handle Problem Language Result Execution time Memory
1103499 2024-10-21T04:38:29 Z ezzzay Nile (IOI24_nile) C++17
32 / 100
 64 ms 14284 KB 
#include<bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
#define pb push_back
#define ll long long
const int N=3e5+5;
int par[N];
int sbtr[N];
int H;
int mn[N];
int find(int x){
	if(par[x]==x)return x;
	return par[x]=find(par[x]);
}
void unite(int x, int y){
	int px=find(x),py=find(y);
	if(px==py)return;
	if(sbtr[py]>sbtr[px])swap(px,py);
	
	if(sbtr[py]%2 and sbtr[px]%2){
		H-=mn[py];
		H-=mn[px];
	}
	else if(sbtr[py]%2){
		H+=mn[py];
		H-=max(mn[py],mn[px]);
	}
	else if(sbtr[px]%2){
		H+=mn[px];
		H-=max(mn[py],mn[px]);
	}
	
	sbtr[px]+=sbtr[py];
	par[py]=px;
	mn[py]=min(mn[py],mn[px]);
	
}
std::vector<long long> calculate_costs(std::vector<int> W, std::vector<int> A,
                                       std::vector<int> B, std::vector<int> E) {
    H=0;                                   	
	for(auto p:A)H+=p;                   	
    int q = (int)E.size();
    int n = W.size();
    for(int i=0;i<n;i++){
		par[i]=i;
		sbtr[i]=1;
		mn[i]=A[i]-B[i];
	}
	
    vector<ll>ans(q);
    vector<pair<ll,ll> >v;
    vector<pair<ll, pair<int,int> > >vc;
    for(int i=0;i<n;i++){
    	v.pb({W[i],i});
	}
	sort(v.begin(),v.end());
	for(int i=1;i<n;i++){
		int x=abs(v[i].ff-v[i-1].ff);
		int a=v[i-1].ss,b=v[i].ss;
		vc.pb({x,{a,b}});
	}
	sort(vc.begin(),vc.end());
	int idx=0;
	vector<pair<int,int>>vec;
	for(int i=0;i<q;i++){
		vec.pb({E[i],i});
	}
	sort(vec.begin(),vec.end());
	
	for(int f=0;f<q;f++){
		ll d=vec[f].ff;
		int k=vec[f].ss;
		while(vc[idx].ff<=d and idx<vc.size()){
			int a=vc[idx].ss.ff,b=vc[idx].ss.ss;
			unite(a,b);
			idx++;
		}
		ans[k]=H;
	}
	return ans;
}

Compilation message

nile.cpp: In function 'std::vector<long long int> calculate_costs(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
nile.cpp:74:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   74 |   while(vc[idx].ff<=d and idx<vc.size()){
      |                           ~~~^~~~~~~~~~

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2384 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #1
0.0 / 6.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2640 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0.0 / 13.0

# Verdict Execution time Memory Grader output
1 Incorrect 28 ms 11448 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 11460 KB Output is correct
2 Correct 36 ms 11456 KB Output is correct
3 Correct 37 ms 11452 KB Output is correct
4 Correct 49 ms 11452 KB Output is correct
5 Correct 39 ms 11436 KB Output is correct
6 Correct 39 ms 12096 KB Output is correct

Subtask #4
0.0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2640 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0.0 / 20.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2640 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 11460 KB Output is correct
2 Correct 36 ms 11456 KB Output is correct
3 Correct 37 ms 11452 KB Output is correct
4 Correct 49 ms 11452 KB Output is correct
5 Correct 39 ms 11436 KB Output is correct
6 Correct 39 ms 12096 KB Output is correct
7 Correct 55 ms 12984 KB Output is correct
8 Correct 60 ms 12996 KB Output is correct
9 Correct 60 ms 12992 KB Output is correct
10 Correct 59 ms 12984 KB Output is correct
11 Correct 60 ms 13248 KB Output is correct
12 Correct 59 ms 12992 KB Output is correct
13 Correct 58 ms 12872 KB Output is correct
14 Correct 59 ms 14284 KB Output is correct
15 Correct 64 ms 14008 KB Output is correct

Subtask #7
0.0 / 18.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2384 KB Output isn't correct
2 Halted 0 ms 0 KB -