保龄球python

classic12332

于 2025-02-27 18:25:16 发布

阅读量579

点赞数 13

文章标签： python 算法开发语言

本文链接：https://blog.csdn.net/qq_45997271/article/details/145910964

版权

P1918 保龄球

题目描述

DL 算缘分算得很烦闷，所以常常到体育馆去打保龄球解闷。因为他保龄球已经打了几十年了，所以技术上不成问题，于是他就想玩点新花招。

DL 的视力真的很不错，竟然能够数清楚在他前方十米左右每个位置的瓶子的数量。他突然发现这是一个炫耀自己好视力的借口——他看清远方瓶子的个数后从某个位置发球，这样就能打倒一定数量的瓶子。

$\bigcirc \bigcirc \bigcirc$
$\bigcirc \bigcirc \bigcirc\ \bigcirc$
$\bigcirc$
$\bigcirc\ \bigcirc$

如上图，每个 “ $\bigcirc$ ” 代表一个瓶子。如果 DL 想要打倒 $3$ 个瓶子就在 $1$ 位置发球，想要打倒 $4$ 个瓶子就在 $2$ 位置发球。

现在他想要打倒 $m$ 个瓶子。他告诉你每个位置的瓶子数，请你给他一个发球位置。

输入格式

第一行包含一个正整数 $n$ ，表示位置数。

第二行包含 $n$ 个正整数 $a_i$ ，表示第 $i$ 个位置的瓶子数，保证各个位置的瓶子数不同。

第三行包含一个正整数 $Q$ ，表示 DL 发球的次数。

第四行至文件末尾，每行包含一个正整数 $m$ ，表示 DL 需要打倒 $m$ 个瓶子。

输出格式

共 $Q$ 行。每行包含一个整数，第 $i$ 行的整数表示 DL 第 $i$ 次的发球位置。若无解，则输出 $0$ 。

输入输出样例 #1

输入 #1

输出 #1

3
0

说明/提示

【数据范围】

对于 $50\%$ 的数据， $\leq n, Q \leq 1000, 1 \leq a_i, m \leq 10^5$ 。

对于 $100\%$ 的数据， $\leq n,Q \leq 100000, 1 \leq a_i, m \leq 10^9$ 。

import sys

def main():
    # 使用 sys.stdin.read 一次性读取所有输入
    input = sys.stdin.read
    data = input().split()
    
    # 处理输入数据
    index = 0
    
    n = int(data[index])  # 位置数
    index += 1
    
    a = list(map(int, data[index:index + n]))  # 每个位置的瓶子数
    index += n
    
    Q = int(data[index])  # 发球次数
    index += 1
    
    # 构建瓶子数到位置的映射
    bottle_to_position = {a[i]: i + 1 for i in range(n)}  # 哈希表，a[i]是瓶子数，i+1是位置（1-based）
    
    # 收集查询结果
    results = []
    for i in range(Q):
        m = int(data[index])  # 需要打倒的瓶子数
        index += 1
        # 查找对应位置
        if m in bottle_to_position:
            results.append(str(bottle_to_position[m]))
        else:
            results.append('0')
    
    # 一次性输出所有结果
    sys.stdout.write("\n".join(results) + "\n")

# 调用主函数
if __name__ == "__main__":
    main()