洛谷【深基15.例2】寄包柜

题目描述

超市里有 n(n\le10^5)n(n≤10
5
) 个寄包柜。每个寄包柜格子数量不一，第 ii 个寄包柜有 a_i(a_i\le10^5)a
i
​
(a
i
​
≤10
5
) 个格子，不过我们并不知道各个 a_ia
i
​
的值。对于每个寄包柜，格子编号从 1 开始，一直到 a_ia
i
​
。现在有 q(q\le10^5)q(q≤10
5
) 次操作：

1 i j k：在第 ii 个柜子的第 jj 个格子存入物品 k(0\le k\le 10^9)k(0≤k≤10
9
)。当 k=0k=0 时说明清空该格子。
2 i j：查询第 ii 个柜子的第 jj 个格子中的物品是什么，保证查询的柜子有存过东西。
已知超市里共计不会超过 10^710
7
个寄包格子，a_ia
i
​
是确定然而未知的，但是保证一定不小于该柜子存物品请求的格子编号的最大值。当然也有可能某些寄包柜中一个格子都没有。

输入格式
第一行 2 个整数 n 和 q，寄包柜个数和询问次数。

接下来 q 个整数，表示一次操作。

输出格式
对于查询操作时，输出答案。

输入输出样例
输入 #1 复制
5 4
1 3 10000 114514
1 1 1 1
2 3 10000
2 1 1
输出 #1 复制
114514
1

---

查询操作，还是要看unordered_map，一个基于哈希表的存在，这就是STL

第一法，将a与b映射为一个大数

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <unordered_map> 

using namespace std;

typedef long long ll;

const int N = 100010;

int n, q;
unordered_map<ll, ll> heap;

int main(){
    cin >> n >> q;
    while (q -- ){
        int x;
        scanf("%d", &x);
        if (x == 1){
            ll a, b, w;
            scanf("%lld%lld%lld", &a, &b, &w);
            heap[a*N + b] = w;
        }else if (x == 2){
            ll a, b;
            scanf ("%lld%lld", &a, &b);
            cout << heap[a*N + b] << endl;
        }
    }
    
    return 0;
}

第二法，二维映射

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <unordered_map> 

using namespace std;

typedef pair<int, int> PII;
const int N = 100010;

int n, q;
//建立二维映射
unordered_map<int, unordered_map<int, int>> heap;

int main(){
    cin >> n >> q;
    while (q -- ){
        int x;
        scanf("%d", &x);
        if (x == 1){
            int a, b, w;
            scanf("%d%d%d", &a, &b, &w);
            heap[a][b] = w;
        }else if (x == 2){
            int a, b;
            scanf ("%d%d", &a, &b);
            cout << heap[a][b] << endl;
        }
    }
    
    return 0;
}