{ }codelper
泛型更新字典P0384

两边都能查的配对账本

学校用两本账本来记配对。
储物柜账本(locker)把储物柜编号(整数)和学生名字配成一对,绰号账本(nick)把绰号和学生名字配成一对。
一对里的编号或绰号叫作左边,名字叫作右边。
在同一本账本里,一个左边最多只在一对里,一个右边也最多只在一对里。
两本账本互不影响,一开始都是空的。
命令的形式是“账本 动作 [值]”,账本是 locker 或 nick。
- put a b:如果有左边是 a 的一对就删掉,如果有右边是 b 的一对也删掉,然后把 a 和 b 新配成一对。
- get a:输出左边是 a 的那一对的右边。
没有这样的一对就输出 -。
- find b:输出右边是 b 的那一对的左边。
没有这样的一对就输出 -。
- del a:如果有左边是 a 的一对就删掉。
- drop b:如果有右边是 b 的一对就删掉。
- size:输出这本账本里的对数。
第一行给出命令数 q,接下来 q 行每行一条命令。
每个 get、find、size 输出一行。
只写一个把左边的类型 K 和右边的类型 V 作为参数的泛型类(C++ 是类模板)BiMap<K, V>,两本账本都用这个类来做。
再写一个方法 inverse(),返回把同样这些配对反过来看的账本。
它的类型是 BiMap<V, K>。

输入
1 ≤ q ≤ 20000。
储物柜编号是 0 到 1000000000 之间的整数,前面没有多余的 0。
绰号和名字由 1~10 个英文小写字母组成。
一行里的词用一个空格隔开。
get、find、size 中至少给出一个
输出
每个 get、find、size 输出一行,按命令出现的顺序
样例 1 · stdin
18
locker put 7 ann
locker put 0 bo
nick put ace ann
locker get 7
locker put 7 bo
locker find ann
locker find bo
locker get 0
locker size
locker put 0 cy
locker find cy
locker get 3
locker find dan
locker size
nick find ann
nick drop ann
nick get ace
nick size
stdout
ann
-
7
-
1
0
-
-
2
ace
-
0
main.py
输入(stdin)

运行时这些行会依次作为 input()/标准输入传入(运行中不会等待输入)。

$ 就绪

请在电脑上打开以编写和运行代码。在手机上可以查看题目和题解。

其他语言怎么写?查看各语言题解 · 4 种语言题解

这道题只接受有泛型的 TypeScript、Java、C# 和 C++(C++ 是类模板)。
Python 和 JavaScript 不写类型,所以一个类从一开始就什么都能装(Python 的 Generic[K, V] 只是写着,运行时不检查);C 和 Craykon 没有类型参数。
BiMap<K, V> 同时修改两个字典(K→V 和 V→K)。
inverse() 返回把这两个字典换了位置拿着的 BiMap<V, K>,所以 find b 就是 inverse().get(b),drop b 就是 inverse().remove(b)。
要点是类型参数有两个,而且它们的位置会互换。
类型实参的写法是 TS BiMap.create<number, string>(),Java new BiMap<Integer, String>(),C# new BiMap<int, string>(),C++ BiMap<int, string>。
Java 写 BiMap<int, String> 是编译错误(unexpected type),所以用 Integer 来装。
只有左边类型不同的两本账本用一个泛型函数 run<K> 处理,同时传入把字符串变成 K 的函数。
返回“没有”的办法各语言不同。
TS 用 V | undefined,Java 用 null,C# 用 bool TryGet(K k, out V v),C++ 用 optional<V>。
陷阱一:如果 put 不删掉旧的配对,旧的配对就会留下来。
例子里 locker put 7 bo 之后,locker find ann 得到 7,locker get 0 得到 bo,locker size 得到 2。
陷阱二:在 TS 里用 x ? String(x) : "-" 来判断“没有”,数字 0 也算假,所以例子里的 locker find cy 不是 0 而是 -。
要用 x === undefined 来问。
陷阱三:在 C# 里如果没有就返回 default(V),再看是不是 null,那么 V 是 int 时 default 是 0。
例子里的 locker find ann 和 locker find dan 输出 0,和真正的 0 号储物柜分不开。
陷阱四:Java 把 get 返回的 null 用 String.valueOf 放进字符串,输出的就不是 - 而是 null。
陷阱五:用 C++ map 的 operator[] 来查,不存在的键会带着默认值被建进字典。
把默认值当作“没有”的解法,在例子里 locker size 会涨成 2 和 3,locker find cy 也变成 -。
要用 find 来查。

TypeScriptmain.ts · 68 行
import { readFileSync } from "node:fs"; // 왼쪽 K와 오른쪽 V를 하나씩 짝짓는 두 방향 사전class BiMap<K, V> {  private constructor(    private readonly fwd: Map<K, V>,    private readonly rev: Map<V, K>,  ) {}   static create<K, V>(): BiMap<K, V> {    return new BiMap<K, V>(new Map<K, V>(), new Map<V, K>());  }   // 같은 짝들을 거꾸로 보는 사전: 타입 인자의 자리가 바뀐다  inverse(): BiMap<V, K> {    return new BiMap<V, K>(this.rev, this.fwd);  }   get(k: K): V | undefined {    return this.fwd.get(k);  }   remove(k: K): void {    const v = this.fwd.get(k);    if (v === undefined) return;    this.fwd.delete(k);    this.rev.delete(v);  }   put(k: K, v: V): void {    this.remove(k);    this.inverse().remove(v);    this.fwd.set(k, v);    this.rev.set(v, k);  }   get size(): number {    return this.fwd.size;  }} function show<T>(x: T | undefined): string {  return x === undefined ? "-" : String(x);} const out: string[] = []; // 왼쪽 타입 K만 다른 두 장부를 같은 코드로 처리한다function run<K>(m: BiMap<K, string>, parse: (s: string) => K, p: string[]): void {  const cmd = p[1];  if (cmd === "put") m.put(parse(p[2]), p[3]);  else if (cmd === "get") out.push(show(m.get(parse(p[2]))));  else if (cmd === "find") out.push(show(m.inverse().get(p[2])));  else if (cmd === "del") m.remove(parse(p[2]));  else if (cmd === "drop") m.inverse().remove(p[2]);  else out.push(String(m.size));} const lines: string[] = readFileSync(0, "utf8").split("\n");const q = Number(lines[0]);const locker = BiMap.create<number, string>();const nick = BiMap.create<string, string>();for (let i = 1; i <= q; i++) {  const p = lines[i].trim().split(/\s+/);  if (p[0] === "locker") run(locker, Number, p);  else run(nick, (s) => s, p);}console.log(out.join("\n"));
JavaMain.java · 67 行
import java.io.*;import java.util.*;import java.util.function.Function; // 왼쪽 K와 오른쪽 V를 하나씩 짝짓는 두 방향 사전class BiMap<K, V> {    private final Map<K, V> fwd;    private final Map<V, K> rev;     BiMap() { this(new HashMap<>(), new HashMap<>()); }     private BiMap(Map<K, V> fwd, Map<V, K> rev) {        this.fwd = fwd;        this.rev = rev;    }     // 같은 짝들을 거꾸로 보는 사전: 타입 인자의 자리가 바뀐다    BiMap<V, K> inverse() { return new BiMap<>(rev, fwd); }     // 없으면 null    V get(K k) { return fwd.get(k); }     void remove(K k) {        V v = fwd.remove(k);        if (v != null) rev.remove(v);    }     void put(K k, V v) {        remove(k);        inverse().remove(v);        fwd.put(k, v);        rev.put(v, k);    }     int size() { return fwd.size(); }} public class Main {    static StringBuilder sb = new StringBuilder();     static <T> String show(T x) { return x == null ? "-" : x.toString(); }     // 왼쪽 타입 K만 다른 두 장부를 같은 코드로 처리한다    static <K> void run(BiMap<K, String> m, Function<String, K> parse, String[] p) {        switch (p[1]) {            case "put": m.put(parse.apply(p[2]), p[3]); break;            case "get": sb.append(show(m.get(parse.apply(p[2])))).append('\n'); break;            case "find": sb.append(show(m.inverse().get(p[2]))).append('\n'); break;            case "del": m.remove(parse.apply(p[2])); break;            case "drop": m.inverse().remove(p[2]); break;            default: sb.append(m.size()).append('\n');        }    }     public static void main(String[] args) throws IOException {        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));        int q = Integer.parseInt(br.readLine().trim());        BiMap<Integer, String> locker = new BiMap<>(); // int는 타입 인자가 될 수 없어 Integer로 담는다        BiMap<String, String> nick = new BiMap<>();        for (int i = 0; i < q; i++) {            String[] p = br.readLine().trim().split(" ");            if (p[0].equals("locker")) run(locker, Integer::valueOf, p);            else run(nick, s -> s, p);        }        System.out.print(sb);    }}
C#Main.cs · 79 行
using System;using System.Collections.Generic;using System.Text; // 왼쪽 K와 오른쪽 V를 하나씩 짝짓는 두 방향 사전class BiMap<K, V>{    private readonly Dictionary<K, V> fwd;    private readonly Dictionary<V, K> rev;     public BiMap() : this(new Dictionary<K, V>(), new Dictionary<V, K>()) { }     private BiMap(Dictionary<K, V> fwd, Dictionary<V, K> rev)    {        this.fwd = fwd;        this.rev = rev;    }     // 같은 짝들을 거꾸로 보는 사전: 타입 인자의 자리가 바뀐다    public BiMap<V, K> Inverse() => new BiMap<V, K>(rev, fwd);     // 있는지 없는지를 bool로 돌려주고, 값은 out으로 내보낸다    public bool TryGet(K k, out V v) => fwd.TryGetValue(k, out v);     public void Remove(K k)    {        if (fwd.Remove(k, out V v)) rev.Remove(v);    }     public void Put(K k, V v)    {        Remove(k);        Inverse().Remove(v);        fwd[k] = v;        rev[v] = k;    }     public int Count => fwd.Count;} class Program{    static readonly StringBuilder sb = new StringBuilder();     // 왼쪽 타입 K만 다른 두 장부를 같은 코드로 처리한다    static void Run<K>(BiMap<K, string> m, Func<string, K> parse, string[] p)    {        switch (p[1])        {            case "put": m.Put(parse(p[2]), p[3]); break;            case "get":                if (m.TryGet(parse(p[2]), out string name)) sb.Append(name); else sb.Append('-');                sb.Append('\n');                break;            case "find":                if (m.Inverse().TryGet(p[2], out K left)) sb.Append(left); else sb.Append('-');                sb.Append('\n');                break;            case "del": m.Remove(parse(p[2])); break;            case "drop": m.Inverse().Remove(p[2]); break;            default: sb.Append(m.Count).Append('\n'); break;        }    }     static void Main()    {        var lines = Console.In.ReadToEnd().Split('\n');        int q = int.Parse(lines[0].Trim());        var locker = new BiMap<int, string>(); // int를 포장 없이 그대로 담는다        var nick = new BiMap<string, string>();        for (int i = 1; i <= q; i++)        {            var p = lines[i].Trim().Split(' ');            if (p[0] == "locker") Run(locker, int.Parse, p);            else Run(nick, s => s, p);        }        Console.Write(sb);    }}
C++main.cpp · 89 行
#include <iostream>#include <map>#include <memory>#include <optional>#include <string>using namespace std; // 왼쪽 K와 오른쪽 V를 하나씩 짝짓는 두 방향 사전template <class K, class V>class BiMap {    shared_ptr<map<K, V>> fwd_;    shared_ptr<map<V, K>> rev_; public:    BiMap() : fwd_(make_shared<map<K, V>>()), rev_(make_shared<map<V, K>>()) {}    BiMap(shared_ptr<map<K, V>> fwd, shared_ptr<map<V, K>> rev) : fwd_(std::move(fwd)), rev_(std::move(rev)) {}     // 같은 짝들을 거꾸로 보는 사전: 타입 인자의 자리가 바뀐다    BiMap<V, K> inverse() const { return BiMap<V, K>(rev_, fwd_); }     // find는 없는 키를 만들지 않는다    optional<V> get(const K& k) const {        auto it = fwd_->find(k);        if (it == fwd_->end()) return nullopt;        return it->second;    }     void remove(const K& k) {        auto it = fwd_->find(k);        if (it == fwd_->end()) return;        rev_->erase(it->second);        fwd_->erase(it);    }     void put(const K& k, const V& v) {        remove(k);        inverse().remove(v);        (*fwd_)[k] = v;        (*rev_)[v] = k;    }     size_t size() const { return fwd_->size(); }}; template <class T>void show(const optional<T>& x) {    if (x) cout << *x << '\n';    else cout << "-\n";} // 왼쪽 타입 K만 다른 두 장부를 같은 코드로 처리한다template <class K, class Parse>void run(BiMap<K, string>& m, Parse parse, const string& cmd) {    string a, b;    if (cmd == "put") {        cin >> a >> b;        m.put(parse(a), b);    } else if (cmd == "get") {        cin >> a;        show(m.get(parse(a)));    } else if (cmd == "find") {        cin >> b;        show(m.inverse().get(b));    } else if (cmd == "del") {        cin >> a;        m.remove(parse(a));    } else if (cmd == "drop") {        cin >> b;        m.inverse().remove(b);    } else {        cout << m.size() << '\n';    }} int main() {    ios::sync_with_stdio(false);    cin.tie(nullptr);    int q;    cin >> q;    BiMap<int, string> locker;    BiMap<string, string> nick;    for (int i = 0; i < q; i++) {        string ledger, cmd;        cin >> ledger >> cmd;        if (ledger == "locker") run(locker, [](const string& s) { return stoi(s); }, cmd);        else run(nick, [](const string& s) { return s; }, cmd);    }    return 0;}