oj.uz

답안 #623088

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
623088 2022-08-05T07:14:53 Z CSQ31 Road Construction (JOI21_road_construction) C++17
5 / 100
 8691 ms 2097152 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
//consider the case where we only need k smallest from two different set of points
//if can do this in some reasonable time then apply zs bit trick
//+log factor
//need <= log^3 maybe?
//ok two set need log^2(?)
#define all(a) a.begin(),a.end()
#define pb push_back
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
const int MAXN = 2000000;
vector<ll>glo;
multiset<int>fen[MAXN];
multiset<int,greater<int>>::iterator it[MAXN];
vector<int>touch;
void upd(int i,int n,int v){
	for(;i<=n;i+=i&(-i))fen[i].insert(v);
}
void query(int i){
	while(i>0){
		touch.pb(i);
		i-=i&(-i);
	}
}
#define fi first
#define se second 
typedef pair<int,int> pii;
int main()
{
	owo
	int n,k;
	cin>>n>>k;
	vector<array<int,2>>p(n);
	vector<int>cy(n);
	for(int i=0;i<n;i++)cin>>p[i][0]>>p[i][1];
	sort(all(p));
	vector<int>crd;
	for(int i=0;i<n;i++)crd.push_back(p[i][1]);
	sort(all(crd));
	crd.resize(unique(all(crd)) - crd.begin());
	for(int i=0;i<n;i++)cy[i] = lower_bound(all(crd),p[i][1]) - crd.begin()+1;
	int m = crd.size();
	//y(i) >= y(j) case
	//we use x(i)+y(i) - x(j) - y(j)
	for(int i=0;i<n;i++){
		touch.clear();
		query(cy[i]);
		priority_queue<pii,vector<pii>,greater<pii>>pq;
		int s = touch.size();
		for(int i=0;i<s;i++){
			if(fen[touch[i]].empty())continue;
			it[i] = fen[touch[i]].begin();
			pq.push({*it[i],i});
		}
		for(int cnt=0;cnt<k;cnt++){
			if(pq.empty())break;
			pii v = pq.top();
			pq.pop();
			ll c = p[i][0] + p[i][1];
			c-=v.fi;
			glo.push_back(c);
			it[v.se]++;
			if(it[v.se] == fen[touch[v.se]].end())continue;
			pq.push({*it[v.se],v.se});
			
		}
	    upd(cy[i],m,p[i][0]+p[i][1]);
	}
	//y(i) < y(j) case
	//we use x(i) - y(i) - (x(j)-y(j)) 
	for(int i=1;i<=m;i++)fen[i].clear(); //i need to do the opposite directly
	for(int i=0;i<n;i++){
		touch.clear();
		cy[i] = m-cy[i]+1;
		query(cy[i]-1);
		priority_queue<pii,vector<pii>,greater<pii>>pq;
		int s = touch.size();
		for(int i=0;i<s;i++){
			if(fen[touch[i]].empty())continue;
			it[i] = fen[touch[i]].begin();
			pq.push({*it[i],i});
		}
		for(int cnt=0;cnt<k;cnt++){
			if(pq.empty())break;
			pii v = pq.top();
			pq.pop();
			ll c = p[i][0] - p[i][1];
			c-=v.fi;
			glo.push_back(c);
			it[v.se]++;
			if(it[v.se] == fen[touch[v.se]].end())continue;
			pq.push({*it[v.se],v.se});
			
		}
	    upd(cy[i],m,p[i][0]-p[i][1]);
	}
	sort(all(glo));
	//for(int x:glo)cout<<x<<" ";
	//cout<<'\n';
	for(int i=0;i<k;i++)cout<<glo[i]<<'\n';
}

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 120 ms 116580 KB Output is correct
2 Correct 127 ms 116656 KB Output is correct
3 Correct 97 ms 114676 KB Output is correct
4 Correct 93 ms 114764 KB Output is correct
5 Correct 130 ms 115560 KB Output is correct
6 Correct 84 ms 114192 KB Output is correct

Subtask #2
0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 4209 ms 2097152 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #3
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3549 ms 224708 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3549 ms 224708 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 27.0

# 결과 실행 시간 메모리 Grader output
1 Correct 120 ms 116580 KB Output is correct
2 Correct 127 ms 116656 KB Output is correct
3 Correct 97 ms 114676 KB Output is correct
4 Correct 93 ms 114764 KB Output is correct
5 Correct 130 ms 115560 KB Output is correct
6 Correct 84 ms 114192 KB Output is correct
7 Runtime error 8691 ms 2097152 KB Execution killed with signal 9
8 Halted 0 ms 0 KB -

Subtask #6
0 / 35.0

# 결과 실행 시간 메모리 Grader output
1 Correct 120 ms 116580 KB Output is correct
2 Correct 127 ms 116656 KB Output is correct
3 Correct 97 ms 114676 KB Output is correct
4 Correct 93 ms 114764 KB Output is correct
5 Correct 130 ms 115560 KB Output is correct
6 Correct 84 ms 114192 KB Output is correct
7 Runtime error 4209 ms 2097152 KB Execution killed with signal 9
8 Halted 0 ms 0 KB -