447. 瑞士轮 归并

Posted 2021-05-26 幽殇默

https://www.acwing.com/problem/content/description/449/

sort()的朴素做法: 时间复杂度O(RNlongN)

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5*2+20;
struct student
{
	long long int id,s,p;
}stu[N];
bool cmp(student a,student b)
{
	if(a.s==b.s)
		return a.id<b.id;
	return a.s>b.s;
}
int main(void)
{
	int n,r,q; cin>>n>>r>>q;
	for(int i=1;i<=2*n;i++) cin>>stu[i].s,stu[i].id=i;
	for(int i=1;i<=2*n;i++) cin>>stu[i].p;
	sort(stu+1,stu+2*n+1,cmp);
	while(r--)
	{
	    
		for(int i=2;i<=2*n;i+=2)
		{
			if(stu[i].p>stu[i-1].p) stu[i].s++;
			else stu[i-1].s++;
		}
		sort(stu+1,stu+2*n+1,cmp);
		
	}
	
	sort(stu+1,stu+2*n+1,cmp);
	cout<<stu[q].id<<endl;
	return 0;
} 

归并排序时间复杂度： O(NlogN+NR)

用归并排序的关键一点在于，其开始我们排好序了。
那么接下来胜者组也还是有序的，败者组也还是有序的。因为同时加1 和不加其排名是不变的

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5*2+20;
struct student
{
	long long int id,s,p;
}stu[N],a[N],b[N];//总  胜   败
bool cmp(student a,student b)
{
	if(a.s==b.s)
		return a.id<b.id;
	return a.s>b.s;
}
int main(void)
{
	int n,r,q; cin>>n>>r>>q;
	for(int i=1;i<=2*n;i++) cin>>stu[i].s,stu[i].id=i;
	for(int i=1;i<=2*n;i++) cin>>stu[i].p;
	sort(stu+1,stu+2*n+1,cmp);
	while(r--)
	{
		int l=1,r=1;
		for(int i=2;i<=2*n;i+=2)
		{
			if(stu[i-1].p>stu[i].p) stu[i-1].s++,a[l++]=stu[i-1],b[r++]=stu[i];
			else stu[i].s++,a[l++]=stu[i],b[r++]=stu[i-1];
		}
		
		int i=1,j=1,k=1;
		while(i<l&&j<r) 
		{
			if(cmp(a[i],b[j])) stu[k++]=a[i++];
			else stu[k++]=b[j++];
		}
		while(i<l) stu[k++]=a[i++];
		while(j<r) stu[k++]=b[j++];
	}
	
	sort(stu+1,stu+2*n+1,cmp);
	cout<<stu[q].id<<endl;
	return 0;
}