Smart infant STLmap

90 阅读1分钟

有 n n n个小孩,以及固定的 200000 200000 200000个幼儿园 ,最开始给你 n n n个信息,小孩 i i i的成长速度和其所在幼儿园,然后有 q q q次转移,把一个小孩 x x x转移到幼儿园 y y y,设 k j k_j kj​为幼儿园 j j j中最大的一个小孩成长速度,对于每次转移,你需要回答: m i n ( k 1 , k 2 , k 3 . . . k 200000 ) min(k_1,k_2,k_3...k_{200000}) min(k1​,k2​,k3​...k200000​)。

考察对STL容器的运用能力。
200000个map维护200000个幼儿园中的小孩信息。
一个map维护所有幼儿园中最大成长速度信息。

AC代码:

#include <bits/stdc++.h>
using namespace std;

map<int, int> mp[200010];
map<int, int> ans;
struct node
{
    int v, pos;
} a[200010];
int main()
{
    int n, q;
    scanf("%d %d", &n, &q);
    for (int i = 1; i <= n; i++)
        scanf("%d %d", &a[i].v, &a[i].pos);
    for (int i = 1; i <= n; i++)
        mp[a[i].pos][a[i].v]++;
    for (int i = 1; i <= 200000; i++)
    {
        if (!mp[i].size())
            continue;
        ans[mp[i].rbegin()->first]++;
    }
    for (int i = 1; i <= q; i++)
    {
        int num, moveto;
        scanf("%d %d", &num, &moveto);
        int Pos = a[num].pos, Rat = a[num].v;
        if (mp[Pos].size() == 1)
        {
            if (ans[Rat] == 1)
                ans.erase(Rat);
            else
                ans[Rat]--;
            mp[Pos].clear();
        }
        else
        {
            if (mp[Pos][Rat] == 1)
            {
                if (Rat == mp[Pos].rbegin()->first)
                {
                    ans[Rat]--;
                    if (!ans[Rat])
                        ans.erase(Rat);
                    ans[(++mp[Pos].rbegin())->first]++;
                }
                mp[Pos].erase(Rat);
            }
            else
            {
                mp[Pos][Rat]--;
            }
        }
        if (!mp[moveto].size())
            ans[Rat]++;
        else if (mp[moveto].rbegin()->first < Rat)
        {
            ans[mp[moveto].rbegin()->first]--;
            if (ans[mp[moveto].rbegin()->first] == 0)
                ans.erase(mp[moveto].rbegin()->first);
            ans[Rat]++;
        }
        mp[moveto][Rat]++;
        a[num].pos = moveto;
        cout << ans.begin()->first << endl;
    }
    return 0;
}