51nod3234 小明爱配对

3234 小明爱配对

小明现在非常喜欢配对，这天有一队人(有男生也有女生)在小明面前排成了一排，每人都有一个技能值，老师给小明出了一道题目，让他来对这些人进行配对。配对有如下要求：

1. 相邻的男女才能进行配对。

2. 每次优先选择技能值差距最小的一对进行配对，如果不止有一对，那么选出最左边的那一对。

3.选出的人离开队列，空位由后面的人补上，这让原来不相邻的男女可能变成相邻的。

就这样一直选取，直到最后配对出现。这个问题难为住了小明，聪明的你可以帮助小明解决这个问题吗？

输入

第一行包括一个数字n，表示队列中人数的个数；
接下来包括n行，每行一个字符和整数，字符B代表女生，字符G代表男生，整数是这个人技能值。

输出

第一行输出配对的个数k；
接下来k行，按配对的顺序输出每对人的编号，对于每对人，先输出编号小的，再输出编号大的。

数据范围

对于10%的数据，1<= n<= 4；
对于50%的数据，1<= n<= 1024；
对于100%的数据，1<= n<=200000；
1<=元素大小<=10000000。

输入样例

4
B 4
G 2
B 4
G 3

输出样例

2
3 4
1 2

解析：

我们用优先队列，来维护题目要求的操作。

优先队列中放置的是配对信息，配对信息包括技能差值以及要配对的人是谁。优先队列按照技能的差值排序，差值最小的在队列的顶部。

初始把符合条件的配对放入优先队列，当弹出一对之后，向前后维护新的配对编号，直到最后队列为空，即是完成了答案的统计。时间复杂度 O(nlogn)。

需要注意的是，如果 1,2 可以配对 2,3 也可以配对，在 1,2 弹出后，2,3 此时已经失效，我们用一个 bool 数组来维护哪些数已经被处理过了，当优先队列弹出 2,3 时就会发现 2 已经被处理过，因此要忽略 2,3 的配对。

放代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 200000 + 10;
pair < int, int > ans[maxn];
int n, q[maxn], z;
char  s;
bool vis[maxn], f[maxn];

struct node {
    int x, y, z;
    friend bool operator < (node a, node b) {
        if (a.z != b.z)
            return a.z > b.z;
        return a.x > b.x;
    }
};

priority_queue<node, vector<node> > pru;

int main() {
    cin >> n;
    memset(f, false, sizeof(f));
    for (int i = 0; i < n; ++i) {
        cin >> s >> q[i + 1];
        f[i + 1] = s == 'B';
    }

    for (int i = 1; i < n; ++i)
        if (f[i] != f[i + 1])
            pru.push((node) { i, i + 1, abs(q[i] - q[i + 1]) });

    memset(vis, false, sizeof(vis));
    z = 0;

    while (!pru.empty()) {
        int x = pru.top().x;
        int y = pru.top().y;
        pru.pop();
        if (!vis[x] && !vis[y])
        {
            vis[y] = vis[x] = true;
            ans[++z].first = x;
            ans[z].second = y;
            while (x > 0 && vis[x])
                --x;
            while (y <= n && vis[y])
                ++y;
            if (x > 0 && y <= n && f[x] != f[y])
                pru.push((node) { x, y, abs(q[x] - q[y]) });
        }
    }
    cout << z << endl;
    for (int i = 1; i <= z; ++i)
        cout << ans[i].first << ' ' << ans[i].second << endl;
    return 0;
}