如何修正猜数字游戏中peg_b计算逻辑的错误？

我来帮你拆解下问题根源，再给出靠谱的修复方案。

问题出在哪？ #

你当前的peg_b计算逻辑存在重复计数的致命问题：

• 对于secret里每一个不是peg_a的字符，你会遍历整个guess去找匹配项，但完全没标记已经被匹配过的guess字符。这就导致同一个guess字符会被secret里的多个字符反复匹配。
• 拿你的测试用例secret="2211"、guess="1002"举例：
  • secret的第一个2匹配guess的第四个2，pb加1；
  • secret的第二个2又匹配同一个guess的第四个2，pb再加1；
  • secret的第一个1匹配guess的第一个1，pb加1；
  • secret的第二个1又匹配同一个guess的第一个1，pb再加1；
  • 最终错误得到pb=4，但实际上guess里只有1个2和1个1能用于非位置匹配，正确结果应该是2。

修复方案：计数统计法 #

我们可以用更高效且准确的统计思路来计算peg_b，步骤如下：

1. 先计算peg_a，同时跳过这些位置，分别统计secret和guess剩余字符的数字出现次数；
2. 对每个数字，取两个统计结果里的较小值，求和就是peg_b（因为每个数字的匹配次数受限于出现次数更少的那一方）。

修复后的代码如下：

#include <string.h>

void get_score(char* secret, char* guess, int* peg_a, int* peg_b) {
    int pa = 0;
    int secret_count[10] = {0}; // 统计0-9在secret非peg_a位置的出现次数
    int guess_count[10] = {0};  // 统计0-9在guess非peg_a位置的出现次数
    
    int len = strlen(secret); // 假设secret和guess长度一致，符合游戏场景
    
    // 第一步：计算peg_a，同时统计非匹配位置的字符
    for (int i = 0; i < len; i++) {
        if (secret[i] == guess[i]) {
            pa++;
        } else {
            secret_count[secret[i] - '0']++;
            guess_count[guess[i] - '0']++;
        }
    }
    
    // 第二步：计算peg_b，取每个数字出现次数的最小值求和
    int pb = 0;
    for (int i = 0; i < 10; i++) {
        pb += (secret_count[i] < guess_count[i]) ? secret_count[i] : guess_count[i];
    }
    
    *peg_a = pa;
    *peg_b = pb;
}

验证你的测试用例 #

对于secret="2211"、guess="1002"：

• 首先计算peg_a：没有位置相同的字符，所以pa=0；
• 统计剩余字符：
  • secret_count中：2出现2次，1出现2次；
  • guess_count中：1出现1次，2出现1次；
• 每个数字取最小值求和：min(2,1) + min(2,1) = 1+1=2，正确得到pb=2。

这个方法不仅解决了重复计数问题，时间复杂度也从原来的O(n²)降到了O(n)，效率更高。

内容的提问来源于stack exchange，提问作者M3Lee OP