洛谷 B3867 [GESP202309 三级] 小杨的储蓄

最新推荐文章于 2025-12-20 17:21:30 发布

原创最新推荐文章于 2025-12-20 17:21:30 发布 · 296 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#算法

B3867 [GESP202309 三级] 小杨的储蓄

https://www.luogu.com.cn/problem/B3867

题目描述

小杨共有 N 个储蓄罐，编号从 0 到 N-1。从第 1 天开始，小杨每天都会往存钱罐里存钱。具体来说，第 i 天他会挑选一个存钱罐 ai，并存入 i 元钱。过了 D 天后，他已经忘记每个储蓄罐里都存了多少钱了，你能帮帮他吗？

输入格式

输入 2 行，第一行两个整数 N,D；第二行 D 个整数，其中第 i 个整数为 ai（保证 0<= ai <=N-1）。

每行的各个整数之间用单个空格分隔。

保证 1≤N≤1,000；1≤D≤1,000。

输出格式

输出 N 个用单个空格隔开的整数，其中第 i 个整数表示编号为 i−1 的存钱罐中有多少钱（i=1,⋯,N）。

输入输出样例

输入 #1

2 3
0 1 0

输出 #1

4 2

输入 #2

3 5
0 0 0 2 0

输出 #2

11 0 4

说明/提示

样例解释 1：

小杨在第 1 天、第 2 天、第 3 天分别向 0 号、 1 号、 0 号存钱罐存了 1 元钱、 2 元钱、 3 元钱，因此 0 号存钱罐有 1+3=4 元钱，而 1 号存钱罐有 2 元钱。

算法分析

1.循环变量i：i从 1 到 d，代表「第 i 天」（题目要求第 i 天存入 i 元）。。
2.读取储蓄罐编号a：每天选择的储蓄罐编号（范围 0~n-1，题目保证输入合法）。
3.核心累加操作：savings[a] += i 等价于 savings[a] = savings[a] + i。
例如样例 1：
第 1 天（i=1），a=0 → savings[0] = 0 + 1 = 1；
第 2 天（i=2），a=1 → savings[1] = 0 + 2 = 2；
第 3 天（i=3），a=0 → savings[0] = 1 + 3 = 4；
最终 savings[0]=4，savings[1]=2，完全符合样例 1 的计算逻辑。

AC代码

#include <iostream>
using namespace std;

int main() {
    int N, D;
    cin >> N >> D;
    
    // 静态数组：大小为N，初始值默认为随机值，需手动初始化
    int savings[N] = {0};  // 局部数组初始化：仅第一个元素为0，其余需循环置0（或用memset）
    // 更稳妥的初始化方式（兼容所有编译器）：
    // for (int i = 0; i < N; ++i) savings[i] = 0;
    
    for (int i = 1; i <= D; ++i) {
        int a;
        cin >> a;
        savings[a] += i;  // 直接通过索引累加
    }
    
    // 输出
    for (int i = 0; i < N; ++i) {
        if (i > 0) cout << " ";
        cout << savings[i];
    }
    cout << endl;
    
    return 0;
}