PTA 1114 全素日 C++实现 简易代码

 

以上图片来自新浪微博，展示了一个非常酷的“全素日”：2019年5月23日。即不仅20190523本身是个素数，它的任何以末尾数字3结尾的子串都是素数。

本题就请你写个程序判断一个给定日期是否是“全素日”。

输入格式：

输入按照 yyyymmdd 的格式给出一个日期。题目保证日期在0001年1月1日到9999年12月31日之间。

输出格式：

从原始日期开始，按照子串长度递减的顺序，每行首先输出一个子串和一个空格，然后输出 Yes，如果该子串对应的数字是一个素数，否则输出 No。如果这个日期是一个全素日，则在最后一行输出 All Prime!。

输入样例 1：

20190523

输出样例 1：

20190523 Yes
0190523 Yes
190523 Yes
90523 Yes
0523 Yes
523 Yes
23 Yes
3 Yes
All Prime!

输入样例 2：

20191231

输出样例 2：

20191231 Yes
0191231 Yes
191231 Yes
91231 No
1231 Yes
231 No
31 Yes
1 No

解题思路： 

取出对应字串判断是否是素数即可

代码示例：

#include <bits/stdc++.h>

using namespace std;

//素数判断
bool isPrime(int num) {
	if (num <= 1) {
		return false;
	}

	for (int i = 2; i <= sqrt(num); i++) {
		if (num % i == 0) {
			return false;
		}
	}
	return true;
}

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

	bool flag = true;
	for (int i = 0; i < date.length(); i++) {
		//从date的第i号位置开始往后取所有的字符给date_son
		string date_son = date.substr(i);

		int num = stoi(date_son);

		if (isPrime(num)) {
			cout << date_son << " " << "Yes\n";
		}
		else {
			cout << date_son << " " << "No\n";
			flag = false;
		}
	}

	if (flag) {
		cout << "All Prime!\n";
	}

	return 0;
}

运行结果：