【Luogu P1878】舞蹈课
Luogu P1878
事实上这道题并不难,但我真没弄懂我手写堆为什么过不了。所以
STL大法好!!!
基本思路
对于每一对相邻异性,将他们的舞蹈技术的差插入一个堆
通过维护这个小根堆,每次就可以取得舞蹈技术差最小的一对
值得注意的是,每次取完一对舞伴之后,要对这对舞伴进行标记,并将堆中所有有这两位舞者参与的舞伴弹出这个堆。并且还需要找到这一对舞伴两边的第一个未被挑出的人,如果是异性则可以作为新的舞伴加入堆
代码
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct data
{
int l,r,val;
bool operator <(const data&x) const
{
if (x.val!=val) return x.val<val;
else return x.l<l;
}//重载运算符
};
int len,n,que[200005],a[200005],ans,group1[200005],group2[200005],rlen;
bool choice[200005];
priority_queue <data> heap;
int main()
{
scanf("%d\n",&n);
for (int i=1;i<=n;i++)
{
char c;
scanf("%c",&c);
if (c=='B') que[i]=true;//对男生进行标记
}
scanf("%d",&a[1]);
for (int i=2;i<=n;i++)
{
scanf("%d",&a[i]);
if (que[i]!=que[i-1]) //可以组成舞伴则进堆
{
data rec;
rec.l=i-1;rec.r=i;
rec.val=abs(a[i]-a[i-1]);
heap.push(rec);
}
}
while (!heap.empty())
{
data rec=heap.top();
ans++;
group1[ans]=rec.l;
group2[ans]=rec.r;
choice[rec.l]=choice[rec.r]=true;//打上标记
while (choice[rec.l]||choice[rec.r])
{
if (heap.empty()) break;
heap.pop();
rec=heap.top();
}//这一个循环的目的就是使堆顶元素是没有被选过的
int l=group1[ans];int r=group2[ans];
while (l>0&&choice[l]) l--;
while (r<=n&&choice[r]) r++;
//寻找两边未被选的人
if (que[l]!=que[r]&&l>0&&r<=n)
{
rec.l=l;
rec.r=r;
rec.val=abs(a[l]-a[r]);
heap.push(rec);
}
}
printf("%d\n",ans);
for (int i=1;i<=ans;i++) printf("%d %d\n",group1[i],group2[i]);
return 0;
}