Sockets CF732E set map

题目：http://codeforces.com/problemset/problem/732/E

题目大意：

有n台电脑（computer）和m个插座（socket），两者均有一个power值，分别为 ai 和 bi ，只有当两者的power值相等时才可以匹配成功。当然，还有足够多个适配器（adapters）可用，插座每接一个适配器，bi 就会减半（取上整）。求最多可匹配多少电脑和插座，以及该种情况下所需要的最少的适配器数量。如有多种方案，输出其中一种即可。

思路：

因为这个题的输出都要用到下标（或编号），所以分别用multimap或pair存储电脑或插座的power值和对应的下标，从最小的插座power开始，匹配对应的电脑power，如果存在则记录，如果无匹配，则将插座power折半继续匹配直到值为1都未匹配成功，那么就说明该插座无论如何都无法匹配到电脑。这样可以使得当前情况下能够匹配到最大的电脑数量而且用最少的适配器，至于查找匹配自然不能遍历，可以将电脑的power存入set容器中进行find查找，效率会高一点，搜索复杂度为logN。

代码如下：

#include<iostream>
#include<algorithm>
#include<utility>
#include<set>
#include<map>
using namespace std;

const int MAX = 2e5 + 2;

multimap<int, int> P;           //电脑与下标对应
multiset<int> S;                //存放电脑power值，用set方便查找
pair<int, int> Sockets[MAX];    //插座与下标对应
int RESULT1[MAX], RESULT2[MAX];

int main()
{
    int N, M, pi, si, c = 0, u = 0;
    cin >> N >> M;
    for (int i = 1; i <= N; ++i)
    {
        cin >> pi;
        S.insert(pi);
        P.insert(make_pair(pi, i));
    }
    for (int i = 1; i <= M; ++i)
    {
        cin >> Sockets[i].first;
        Sockets[i].second = i;
    }
    sort(Sockets + 1, Sockets + M + 1);

    for (int i = 1, num = 0; i <= M; ++i, num = 0)
        while (1)
        {
            multiset<int>::iterator set_it = S.find(Sockets[i].first);
            if (set_it != S.end())
            {
                multimap<int, int>::iterator map_it = P.find(Sockets[i].first);
                RESULT1[Sockets[i].second] = num;
                RESULT2[(*map_it).second] = Sockets[i].second;
                c++; u += num;
                S.erase(set_it), P.erase(map_it);
                break;
            }
            if (Sockets[i].first == 1)break;
            num++;
            Sockets[i].first = (Sockets[i].first + 1) / 2;
        }
    cout << c << " " << u << endl;
    for (int i = 1; i <= M; ++i)cout << RESULT1[i], i == M ? cout << endl : cout << " ";
    for (int i = 1; i <= N; ++i)cout << RESULT2[i], i == N ? cout << endl : cout << " ";
}

感谢您的阅读，生活愉快~