AB23 kotori和素因子

AB23 kotori和素因子

描述
kotori拿到了一些正整数。她决定从每个正整数取出一个素因子。但是，kotori有强迫症，她不允许两个不同的正整数取出相同的素因子。

她想知道，最终所有取出的数的和的最小值是多少？

注：若 a\bmod k== 0amodk==0，则称 kk 是 aa 的因子。若一个数有且仅有两个因子，则称其是素数。显然1只有一个因子，不是素数。
输入描述：
第一行一个正整数 nn ，代表kotori拿到正整数的个数。 第二行共有 n 个数
​ ，表示每个正整数的值。 保证不存在两个相等的正整数。
输出描述：
一个正整数，代表取出的素因子之和的最小值。若不存在合法的取法，则输出-1。
示例1
输入：
4
12 15 28 22
输出：
17
说明：
分别取3，5，7，2，可保证取出的数之和最小

分析：

dfs遍历和判断素数

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
const int N=1e6+10;
ll t,m,n,sum,ma,k;
int a[N],v[N];
int mi=INF;
bool prime(int x){//判断是否为素数 
	for(int i=2;i<=sqrt(x);i++){
		if(x%i==0) return false;
	}
	return true;
}
void dfs(int x,set<int> s,int t){
	if(x==n){//遍历至最后一个数 
		mi=min(mi,t);//找到最小的因子和 
		return;
	}
	for(int i=2;i<=a[x];i++){//遍历找因子 
		if(a[x]%i==0&&prime(i)&&!s.count(i)) {
		//如果因子为素数而且该因子没有遍历过 
			s.insert(i);
			dfs(x+1,s,t+i);//和加上该因子 
			s.erase(i);
		}
	}
}
int main() {
    cin>>n;
    for(int i=0;i<n;i++)
        cin>>a[i];
    set<int> s;//set按照元素键值进行排序，且没有重复排序 
    dfs(0,s,0);
    if(mi==INF) cout<<-1<<endl;
    else cout<<mi<<endl;
    return 0;
}