PATB 1094 判断素数

问题描述

政策原因不贴出描述

输入格式：

输入在第一行给出 2 个正整数，分别是 L（不超过 1000 的正整数，为数字长度）和 K（小于 10 的正整数）。接下来一行给出一个长度为 L 的正整数 N。

输出格式：

在一行中输出 N 中最早出现的 K 位连续数字所组成的素数。如果这样的素数不存在，则输出 404。注意，原始数字中的前导零也计算在位数之内。例如在 200236 中找 4 位素数，0023 算是解；但第一位 2 不能被当成 0002 输出，因为在原始数字中不存在这个 2 的前导零。

输入样例 1：

20 5
23654987725541023819

输出样例 1：

49877

输入样例 2：

10 3
2468024680

输出样例 2：

404

问题分析

处理流程：

用字符串接收输入的数据；对每一位数字向后找 k 位，看这个组成的数字是不是素数，是的话输出不是的话继续找；找到输出答案，找不到输出 404。

处理细节：

• 判断素数：用 judge 函数，i * i <= x 卡 范围不容易超时；遇到 0、1直接返回 false 效率也比较高；
• 每k段数据：从头开始遍历数据段，以 str[i] 为起点找长度为 k 的字符串，如果找不到长度为 k 的数据段那么就输出 404 并结束程序，因为只找一个解而且这个解是最早找到的那个，到找不到了的时候还是如此，只能说明找不到。如果找到了还不是素数，那么把临时计数变量置为 0，输出字符串初始化；如果这 k 段数据是素数，打印结果并结束程序；

代码

#include <iostream>
bool judge(int x) {
    if (x == 1 || x == 0) return false;
    for (int i = 2; i * i <= x; i++) if (x % i == 0) return false;
    return true;
}
int main(){
    int l, k, tmp = 0;
    std::string str, out;
    std::cin >> l >> k;
    std::cin >> str;
    for (int i = 0; i < l; i++) {
          for (int j = i; j < k + i; j++) {
               if (j > str.length()) {std::cout << "404"; return 0;}
               tmp = tmp * 10 + str[j] - '0';
               out += str[j]; // 存储临时变量的字符串
          }
          if (judge(tmp)) {std::cout << out; return 0;}
          else{
               tmp = 0;
               std::string newout;
               out = newout;
          }
    }
    return 0;
}

更简单的方法

可以参考柳神的方法，她用了 substr 函数直接切割字符串，比我查找的方式简单多了。