Submission #380610

#	Time	Username	Problem	Language	Result	Execution time	Memory
380610		mariowong	Examination (JOI19_examination)	C++14	100 / 100	772 ms	25056 KiB

examination

#include <bits/stdc++.h>
using namespace std;
 
struct node{
	long long maths,ict,sum;
	int indx;	
}a[200005];
bool cmp(node a1,node a2){
	if (a1.maths != a2.maths)
	return a1.maths > a2.maths;
	return a1.indx < a2.indx;
}
int n,m,pt;
long long seg[800005]; 
pair <long long,int> d[200005];
long long ct,now,ans[200005];
vector <pair<long long,long long>  > u;
vector <pair<pair<long long,long long>,int>  > q;
void build (int id,int x,int y){
	if (x == y)
	seg[id]=0;
	else
	{
		int mid=(x+y)/2;
		if (seg[id] != 0){
			build(id*2,x,mid);
			build(id*2+1,mid+1,y);
		}
		seg[id]=0;
	}
}
void update(int id,int x,int y,int pos){
	if (x == y)
	seg[id]++;
	else
	{
		int mid=(x+y)/2;
		if (pos <= mid) update(id*2,x,mid,pos);
		else update(id*2+1,mid+1,y,pos);
		seg[id]=seg[id*2]+seg[id*2+1];
	}
}
int query(int id,int x,int y,int l,int r){
	if (l <= x && y <= r)
	return seg[id];
	if (x > r || y < l)
	return 0;
	int mid=(x+y)/2;
	return query(id*2,x,mid,l,r)+query(id*2+1,mid+1,y,l,r);
}
void solve(int x,int y){
	if (x == y);
	else
	{
		int mid=(x+y)/2;
		u.clear(); q.clear();
		build(1,0,ct);
		for (int i=x;i<=mid;i++){
			if (a[i].indx == 0)
			u.push_back(make_pair(a[i].ict,a[i].sum));
		}
		for (int i=mid+1;i<=y;i++){
			if (a[i].indx != 0)
			q.push_back(make_pair(make_pair(a[i].ict,a[i].sum),a[i].indx));
		}
		sort(u.begin(),u.end()); sort(q.begin(),q.end()); 
		reverse(u.begin(),u.end()); reverse(q.begin(),q.end()); 
		pt=0;
		for (int i=0;i<q.size();i++){
			while (pt < (int)u.size() && u[pt].first >= q[i].first.first){
				update(1,0,ct,u[pt].second);
				pt++;
			} 
			ans[q[i].second]+=query(1,0,ct,q[i].first.second,ct);
		}
		solve(x,mid);
		solve(mid+1,y);
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin >> n >> m;
	for (int i=1;i<=n;i++){
		cin >> a[i].maths >> a[i].ict;
		a[i].sum=a[i].maths+a[i].ict;
		d[i]=make_pair(a[i].sum,i);
	}
	for (int i=n+1;i<=n+m;i++){
		cin >> a[i].maths >> a[i].ict >> a[i].sum; a[i].indx=i-n;
		d[i]=make_pair(a[i].sum,i);
	}
	sort(d+1,d+1+n+m); 
	for (int i=1;i<=n+m;i++){
		if (d[i].first != d[i-1].first) 
		ct++;
		a[d[i].second].sum=ct;
	}
	sort(a+1,a+1+n+m,cmp); 
	solve(1,n+m);
	for (int i=1;i<=m;i++){
		cout << ans[i] << "\n";
	}
	return 0;
}	

Compilation message (stderr)

examination.cpp: In function 'void solve(int, int)':
examination.cpp:69:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<long long int, long long int>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   69 |   for (int i=0;i<q.size();i++){
      |                ~^~~~~~~~~

• Subtask 1: 2 / 2
• Subtask 2: 20 / 20
• Subtask 3: 21 / 21
• Subtask 4: 57 / 57