蓝桥杯 算法提高 ADV-237 三进制数位和 枚举 条件判断

算法提高 三进制数位和
时间限制：1.0s 内存限制：256.0MB
问题描述
　　给定L和R，你需要对于每一个6位三进制数（允许前导零），计算其每一个数位上的数字和，设其在十进制下为S。
　　一个三进制数被判断为合法，当且仅当S为质数，或者S属于区间[L,R]。
　　你的任务是给出合法三进制数的个数。
输入格式
　　一行两个非负整数L,R。
输出格式
　　一行一个非负整数表示答案。
样例输入
0 0
样例输出
330
数据规模和约定
　　保证0<=L<R<=12。
提示
　　判断x是否为质数核心代码：for (int i = 2; i * i <= x; ++i) if (x % i == 0) {/你猜？/}

分析：这道题总共分三步：
　　1. 枚举6位3进制数
　　2. 求每一个数位上的数字和n
　　3. 判断n是否是质数，或n是否∈[L, R]。
　　　　这里要注意，判断质数时要考虑0和1。
代码如下：

#include <iostream>
#include <cstring>
using namespace std;

//判断是否质数 
bool zhishu(int num)
{
	if(num == 1 || num == 0) return false;
	for(int i = 2; i * i <= num; i++)
	{
		if(num % i == 0) return false;
	}
	return true;
}

//求6位3进制字符串的和
int sum(char *s)
{
	int temp = 0;
	while(*s)
	{
		temp += *s - '0';
		s++;
	}
	return temp;
}

int main()
{
	int L, R;
	char num[7];
	int n, res = 0;
	
	cin >> L >> R;
	
	memset(num, 0, sizeof(num));
	
	for(num[0] = '0'; num[0] <= '2'; num[0]++)
	{
		for(num[1] = '0'; num[1] <= '2'; num[1]++)
		{
			for(num[2] = '0'; num[2] <= '2'; num[2]++)
			{
				for(num[3] = '0'; num[3] <= '2'; num[3]++)
				{
					for(num[4] = '0'; num[4] <= '2'; num[4]++)
					{
						for(num[5] = '0'; num[5] <= '2'; num[5]++)
						{
							n = sum(num);
							if(zhishu(n) || n >= L && n <= R)
							{
								res++;
							}
						}
					}
				}
			}
		}
	}
	
	cout << res << endl;
	
	return 0;
}