{ }codelper
更新字典键是否存在P0064

商店库存管理

商店里每种商品的库存一开始都是 0。
第一行给出命令数 m,接下来 m 行每行一个命令。
- "add 商品 k":该商品库存加 k。
- "sell 商品 k":如果库存至少为 k,就减去 k;否则(包括从未进过货的商品)库存不变,并输出 "失败"。
- "show 商品":输出该商品当前的库存(从未进过货则为 0)。
处理完所有命令后,在最后一行输出库存还有 1 个及以上的商品种类数。

输入
1 ≤ m ≤ 2000,1 ≤ k ≤ 1000。
商品名为 1 到 12 个小写英文字母
输出
按命令顺序每行输出一个 "失败" 或 show 的结果,最后一行输出仍有库存的商品种类数
样例 1 · stdin
7
add pen 3
add cup 2
sell pen 2
sell cup 5
show pen
show ink
sell cup 2
stdout
失败
1
0
1
main.py
输入(stdin)

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

$ 就绪

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

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

陷阱在于第一次出现的商品。
Python stock[item] += k 会抛 KeyError,C# stock[item] += k 会抛 KeyNotFoundException,Java int have = stock.get(item) 把 null 拆箱成 int 时抛 NullPointerException。
读取时给默认值:Python stock.get(item, 0),Java getOrDefault・merge(item, k, Integer::sum),C# GetValueOrDefault,JS Map 的 get(item) ?? 0。
如果在 JS 空对象 {} 上用 stock[item] || 0,遇到 "constructor" 会拿到 Object.prototype 上的函数,变成奇怪的字符串。
C++ stock[item] += k 在键不存在时会先放入 0,所以直接可用(但只读一下也会创建键)。
最后一行也有陷阱:卖到 0 个的商品仍留在字典里,要只数值为 1 及以上的,而不是 len(stock)。
C 没有字典,只能从头查找名称·数量表。
Craykon 先用 stock.包含(item) 询问 字典(),最后遍历 stock.值列表 计数。

Pythonmain.py · 20 行
import sys lines = sys.stdin.read().split("\n")m = int(lines[0])stock = {}for line in lines[1:m + 1]:    parts = line.split()    cmd, item = parts[0], parts[1]    have = stock.get(item, 0)    if cmd == "add":        stock[item] = have + int(parts[2])    elif cmd == "sell":        k = int(parts[2])        if have >= k:            stock[item] = have - k        else:            print("失败")    else:        print(have)print(sum(1 for v in stock.values() if v > 0))
JavaScriptmain.js · 15 行
const lines = require("fs").readFileSync(0, "utf8").split("\n");const m = Number(lines[0]);const stock = new Map();const out = [];for (let i = 1; i <= m; i++) {  const [cmd, item, k] = lines[i].trim().split(" ");  const have = stock.get(item) ?? 0;  if (cmd === "add") stock.set(item, have + Number(k));  else if (cmd === "sell") {    if (have >= Number(k)) stock.set(item, have - Number(k));    else out.push("失败");  } else out.push(have);}out.push([...stock.values()].filter((v) => v > 0).length);console.log(out.join("\n"));
TypeScriptmain.ts · 16 行
import { readFileSync } from "node:fs";const lines: string[] = readFileSync(0, "utf8").split("\n");const m: number = Number(lines[0]);const stock = new Map<string, number>();const out: (string | number)[] = [];for (let i = 1; i <= m; i++) {  const [cmd, item, k] = lines[i].trim().split(" ");  const have: number = stock.get(item) ?? 0;  if (cmd === "add") stock.set(item, have + Number(k));  else if (cmd === "sell") {    if (have >= Number(k)) stock.set(item, have - Number(k));    else out.push("失败");  } else out.push(have);}out.push([...stock.values()].filter((v) => v > 0).length);console.log(out.join("\n"));
JavaMain.java · 30 行
import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;import java.util.HashMap;import java.util.Map; public class Main {    public static void main(String[] args) throws IOException {        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));        int m = Integer.parseInt(in.readLine().trim());        Map<String, Integer> stock = new HashMap<>();        StringBuilder out = new StringBuilder();        for (int i = 0; i < m; i++) {            String[] p = in.readLine().trim().split(" ");            int have = stock.getOrDefault(p[1], 0);            if (p[0].equals("add")) {                stock.merge(p[1], Integer.parseInt(p[2]), Integer::sum);            } else if (p[0].equals("sell")) {                int k = Integer.parseInt(p[2]);                if (have >= k) stock.put(p[1], have - k);                else out.append("失败").append('\n');            } else {                out.append(have).append('\n');            }        }        long inStock = stock.values().stream().filter(v -> v > 0).count();        out.append(inStock).append('\n');        System.out.print(out);    }}
C#Main.cs · 32 行
using System;using System.Collections.Generic;using System.Linq; class Program{    static void Main()    {        int m = int.Parse(Console.ReadLine());        var stock = new Dictionary<string, int>();        for (int i = 0; i < m; i++)        {            string[] p = Console.ReadLine().Trim().Split(' ');            int have = stock.GetValueOrDefault(p[1]);            if (p[0] == "add")            {                stock[p[1]] = have + int.Parse(p[2]);            }            else if (p[0] == "sell")            {                int k = int.Parse(p[2]);                if (have >= k) stock[p[1]] = have - k;                else Console.WriteLine("失败");            }            else            {                Console.WriteLine(have);            }        }        Console.WriteLine(stock.Values.Count(v => v > 0));    }}
C++main.cpp · 31 行
#include <iostream>#include <string>#include <unordered_map> int main() {    int m;    std::cin >> m;    std::unordered_map<std::string, int> stock;    for (int i = 0; i < m; i++) {        std::string cmd, item;        std::cin >> cmd >> item;        auto it = stock.find(item);        int have = it != stock.end() ? it->second : 0;        if (cmd == "add") {            int k;            std::cin >> k;            stock[item] = have + k;        } else if (cmd == "sell") {            int k;            std::cin >> k;            if (have >= k) stock[item] = have - k;            else std::cout << "失败\n";        } else {            std::cout << have << '\n';        }    }    int inStock = 0;    for (const auto& [item, count] : stock)        if (count > 0) inStock++;    std::cout << inStock << '\n';}
Cmain.c · 40 行
#include <stdio.h>#include <string.h> /* C에는 사전이 없다: 물건 이름과 재고를 작은 표에 두고 처음부터 찾는다 */static char names[2000][13];static int counts[2000];static int size = 0; static int find_or_add(const char *item) {    for (int i = 0; i < size; i++)        if (strcmp(names[i], item) == 0) return i;    strcpy(names[size], item);    counts[size] = 0;    return size++;} int main(void) {    int m, k;    char cmd[5], item[13];    scanf("%d", &m);    for (int i = 0; i < m; i++) {        scanf("%4s %12s", cmd, item);        int at = find_or_add(item);        if (strcmp(cmd, "add") == 0) {            scanf("%d", &k);            counts[at] += k;        } else if (strcmp(cmd, "sell") == 0) {            scanf("%d", &k);            if (counts[at] >= k) counts[at] -= k;            else puts("失败");        } else {            printf("%d\n", counts[at]);        }    }    int in_stock = 0;    for (int i = 0; i < size; i++)        if (counts[i] > 0) in_stock++;    printf("%d\n", in_stock);    return 0;}
Craykonmain.cray · 41 行
m = 控制台.读行().转数字();stock = 字典();循环 (i = 0; i < m; i++){    p = 控制台.读行().去空白().分割();    item = p[1];    have = 0;    如果 (stock.包含(item))    {        have = stock[item];    }    如果 (p[0] == "add")    {        stock[item] = have + p[2].转数字();    }    否则 如果 (p[0] == "sell")    {        k = p[2].转数字();        如果 (have >= k)        {            stock[item] = have - k;        }        否则        {            控制台.输出("失败");        }    }    否则    {        控制台.输出(have);    }}inStock = 0;遍历 (v 在 stock.值列表){    如果 (v > 0)    {        inStock++;    }}控制台.输出(inStock);