考试的时候只想到用优先队列,结果光荣超时,还是下来一想还是用分治吧。
把赢的和输的分成两队,用类似与归并排序的方法放在原数组里面。
#include<cstdio>
#include<iostream>
#include<algorithm>
#define maxn 100005
using namespace std;
struct data
{
int s,p,id;
}a[maxn*2],win[maxn],lose[maxn];
int n,R,Q;
data MAX(data a,data b)
{
if(a.p>b.p) return a;
return b;
}
data MIN(data a,data b)
{
if(a.p>b.p) return b;
return a;
}
bool cmp(data a,data b)
{
if(a.s==b.s) return a.id<b.id;
return a.s>b.s;
}
void read(int &x)
{
x=0;
bool ok=false;
char ch;
ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
while((ch>='0'&&ch<='9')||ch=='-')
{
if(ch=='-') ok=1;
else x=x*10+ch-'0';
ch=getchar();
}
if(ok) x=-x;
}
void in()
{
read(n);read(R);read(Q);
for(int i=1;i<=2*n;i++)
{
read(a[i].s);
a[i].id=i;
}
for(int i=1;i<=2*n;i++) read(a[i].p);
}
void task()
{
sort(a+1,a+2*n+1,cmp);
while(R--)
{
for(int i=1;i<=2*n;i+=2)
{
data u=MAX(a[i],a[i+1]);
data v=MIN(a[i],a[i+1]);
win[(i+1)/2]=(data){u.s+1,u.p,u.id};
lose[(i+1)/2]=v;
}
int i=1,j=1,k=0;
while(i<=n&&j<=n)
{
if(win[i].s>lose[j].s) a[++k]=win[i++];
else if(win[i].s==lose[j].s)
{
if(win[i].id<lose[j].id) a[++k]=win[i++];
else a[++k]=lose[j++];
}
else a[++k]=lose[j++];
}
while(i<=n) a[++k]=win[i++];
while(j<=n) a[++k]=lose[j++];
}
printf("%d",a[Q].id);
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
in();
task();
return 0;
}