PTA甲级考试真题练习59——1059 Prime Factors

题目

思路

质因子问题
所谓质因子分解是指将一个正整数n写成一个或多个质数的乘积的形式。
由于每个质因子都可以不止出现一次，因此不妨定义结构体factor，用来存放质因子及其个数：
对一个正整数n来说，如果它存在1和本身之外的因子，那么一定是在sqrt(n)的左右成对，而这里把这个结论用在“质因子”上面，会得到一个强化结论：对一个正整数来说，如果它存在[2,n]范围内的质因子，要么这些质因子全部小于等于sqrt(n)，要么只存在一个大于sqrt(n)的质因子，而其余质因子全部小于等于sqrt(n)。得到下面思路
① 枚举1~sqrt(n)范围内的所有质因子p，判断p是否是n的因子
如果p是n的因子，那么给fac数组中增加质因子p，并初始化其个数为0。然后，只要p还是n的因子，就让n不断除以p，每次操作令p的个数加1，直到p不再是n的因子为止。
如果p不是n的因子，就直接跳过
② 如果在上面步骤结束后n仍然大于1，说明n有且仅有一个大于sqrt(n)的因子，这是需要把这个质因子加入fac数组，并令其个数为1

坑点

注意num为1的时候输出1的情况

代码

#include <iostream>
#include <string>
#include<algorithm>
#include<vector>
 
using namespace std;
#define nmax 100001
#define inf 999999

struct factor{
	int x;
	int cnt;
};
vector<factor> vec;

void FactorPrime(int num)
{
	int sqrtNum = sqrt((double)num);
	for (int i = 2; i <= sqrtNum; ++i) {
		if (num % i != 0)
			continue;
		factor fac;
		fac.x = i;
		fac.cnt = 0;
		while (num % i == 0) {
			fac.cnt++;
			num /= i;
		}
		vec.emplace_back(fac);
	}
	if (num != 1) {
		factor fac;
		fac.x = num;
		fac.cnt = 1;
		vec.emplace_back(fac);
	}
}

int main()
{
	int num;
	cin >> num;
	if (num == 1) {
		cout << num << "=" << num;
		return 0;
	}
	FactorPrime(num);
	cout << num << "=";
	if (vec[0].cnt > 1)
		cout << vec[0].x << "^" << vec[0].cnt;
	else
		cout << vec[0].x;
	for (int i = 1; i < vec.size(); ++i) {
		if (vec[i].cnt > 1)
			cout << "*"<<vec[i].x << "^" << vec[i].cnt;
		else
			cout <<"*"<< vec[i].x;
	}
	return 0;
}