学校用两本账本来记配对。 储物柜账本(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>。
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
错误列表00F8 ↓ · ⇧F8 ↑
输入(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;}