5034. 配对

5034. 配对 - AcWing题库

n 个人参加一场舞会，每个人的性别（男或女）已知。

在开始跳舞之前，需要先给他们进行配对。

初始时，这 n 个人从左到右排成一队，依次编号为 1∼n。

其中，第 i 个人的舞蹈水平为 ai。

当队列中有男有女时，重复以下过程，进行配对：选出当前队列中彼此相邻且舞蹈水平差异最小的一对男女，让他们组成一对舞伴，并离开队列。

注意：

• 当满足彼此相邻且舞蹈水平差异最小的男女不止一对时，选择最靠左的一对男女进行配对。
• 队伍始终都是连续的，当一对男女离开队列后，剩下的人会紧缩队伍，保证队伍的连贯。

请你计算，一共可以产生多少对配对，并按配对的产生顺序，将所有配对依次输出。

输入格式

第一行包含整数 n。

第二行包含一个长度为 n 的字符串，字符串中的每个字符要么为 B，要么为 G，其中第 i 个字符用来表示第 i 个人的性别，B 表示男，G 表示女。

第三行包含 n 个整数 a1,a2,…,an。

输出格式

第一行输出一个整数 k，表示可以产生的配对数量。

接下来 k 行，其中第 i 行输出第 i 对产生的配对的男女编号，每对编号在输出时，先输出较小编号，再输出较大编号。

数据范围

前 4 个测试点满足 1≤n≤10。
所有测试点满足 1≤n≤，1≤ai≤。

输入样例1：

4
BGBG
4 2 4 3

输出样例1：

2
3 4
1 2

输入样例2：

4
BBGG
4 6 1 5

输出样例2：

2
2 3
1 4

输入样例3：

7
BGBGBGB
5 1 2 5 1 2 1

输出样例3：

3
2 3
1 4
5 6

#include<iostream>
#include<queue>
#include<cmath>
#include<vector>
using namespace std;
typedef pair<int,int> PII;
const int N = 200005;
char sex[N];
int a[N],l[N],r[N];
bool st[N];
int n;
vector<PII> res;
struct Data {
    int d,a,b;
    bool operator <(const Data&t)const {
        if(d!=t.d)return d>t.d;
        return a>t.a;
    }
};
priority_queue<Data>heap;
Data getData(int x,int y) {
    return Data{abs(a[x]-a[y]),x,y};
}
void remove(int x,int y) {
    r[l[x]]=r[y];
    l[r[y]]=l[x];
}
int main(void) {
    cin>>n;
    cin>>&sex[1];
    a[0]=a[n+1]=1e9;
    for(int i=1; i<=n; i++) {
        cin>>a[i];
        l[i]=i-1,r[i]=i+1;
        heap.push(getData(i-1,i));
    }
    heap.push(getData(n,n+1));
    while(heap.size()) {
        Data cur=heap.top();
        heap.pop();
        if(sex[cur.a]==sex[cur.b]) {
            continue;
        }
        if(st[cur.a]||st[cur.b]) {
            continue;
        }
        if(cur.a==0||cur.b==n+1) {
            break;
        }
        res.push_back({cur.a,cur.b});
        st[cur.a]=st[cur.b]=true;
        heap.push(getData(l[cur.a],r[cur.b]));
        remove(cur.a,cur.b);
    }
    cout<<res.size()<<endl;
    for(int i=0; i<res.size(); i++) {
        cout<<res[i].first<<" "<<res[i].second<<endl;
    }
    return 0;
}

1. 定义了一个结构体 Data，包含三个整数变量 d、a、b。并重载了 < 运算符，用于比较两个 Data 类型的对象。
2. 定义了一个函数 getData，根据两个点的 a 值生成一个 Data 类型的对象。
3. 定义了一个函数 remove，用于从链表中删除一个节点。
4. 在 main 函数中，首先读入数据，包括节点数 n 和一个字符数组 sex，以及每个节点的 a 值。
5. 将第一个和最后一个节点的 a 值设为某个大值，然后将每个节点与其相邻的节点相连，并将每对相邻的节点加入到优先队列 heap 中。
6. 将最后一节点与第一个节点之间的边也加入到优先队列中。
7. 然后开始一个循环，每次从优先队列中取出一个元素，如果这对节点属于不同的性别、不在已处理的列表中、不是第一个节点或者最后一个节点，那么就将这对节点进行处理。处理的方式是：将这对节点的左右相邻节点加入到优先队列中，然后将这对节点标记为已处理。
8. 处理完成后，将这对节点及其边从链表中删除。
9. 这个过程会一直持续到优先队列为空，或者已经处理了所有的节点。
10. 最后，输出处理的结果，包括处理的点对数量和所有处理过的点对。