2024年1月24日-秋招-第一题（100分）-计算积分

在线评测链接

题目描述

小明有根绳子，上面有很多玻璃球，玻璃球由红绿蓝三种颜色组成，其中红色球1分，绿色求2分，蓝色球三分。

现在有一个积分规则，当一个球与前面1个球颜色相同会奖励一分，和前面2个球都相同会奖励2分，以此类推。

现在给定一串玻璃球，计算该玻璃球的累计分数，红色玻璃球用字符r表示，绿色用g表示，蓝色用b表示。

输入描述

输入：一个string由r,g,b组成的字符串

输出描述

输出：经过计算后的得分

备注：

• 1 <= string.length <= 10000

示例1

输入

rgb

输出

6

说明：

1+2+3=6

示例2

输入

rrrggb

输出

14

说明：

1+2（奖励1分）+3（奖励2分）+2+3（奖励1分）+3=14

思路：模拟遍历

由于string的长度只有1e4，可以直接暴力做，n^2也只有1e8，小常数1s内可以跑完。直接往前遍历有几个字符和它相同即可。

或者基于简单dp的方式，每个位置保存它前面与它相等且串联的字符数量，每个字符只需要看其是否与上一个字符相等即可，相等则加1，否则置为0。

JavaScript

let s = readline();
let ans = 0;
let score = { "r": 1, "g": 2, "b": 3 };

for (let i = 0; i < s.length; i++) {
    for (let j = i - 1; j >= 0; j--) {
        if (s[i] == s[j]) {
            ans += 1;
        } else {
            break;
        }
    }
    ans += score[s[i]];
}

print(ans);

Java

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        String s = scanner.next();

        int ans = 0;
        Map<Character, Integer> score = new HashMap<>();
        score.put('r', 1);
        score.put('g', 2);
        score.put('b', 3);

        for (int i = 0; i < s.length(); i++) {
            for (int j = i - 1; j >= 0; j--) {
                if (s.charAt(i) == s.charAt(j)) {
                    ans += 1;
                } else {
                    break;
                }
            }
            ans += score.get(s.charAt(i));
        }

        System.out.println(ans);
    }
}

Python

s = input()
ans = 0
score = {"r": 1, "g": 2, "b": 3}
for i in range(len(s)):
    for j in range(i - 1, -1, -1):
        if s[i] == s[j]:
            ans += 1
        else:
            break
    ans += score[s[i]]
print(ans)

C++

#include <iostream>
#include <string>
#include <map>

using namespace std;

int main() {
    string s;
    cin >> s;

    int ans = 0;
    map<char, int> score = { {'r', 1}, {'g', 2}, {'b', 3} };

    for (int i = 0; i < s.length(); i++) {
        for (int j = i - 1; j >= 0; j--) {
            if (s[i] == s[j]) {
                ans += 1;
            } else {
                break;
            }
        }
        ans += score[s[i]];
    }

    cout << ans << endl;

    return 0;
}