17.小M的数组变换 [已解决：集合记录可能存在的素数比较其的种类与数组集合大小对比] C++

超爱笑嘻嘻

于 2025-01-07 16:39:42 发布

阅读量95

点赞数 3

文章标签： c++ 算法开发语言

本文链接：https://blog.youkuaiyun.com/qq_48462433/article/details/144988725

版权

#include <iostream>
#include <unordered_set>
#include <vector>

using namespace std;
string solution(int n, vector<int>& a) {
    // PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
    // write code here
    unordered_set<int> set = {};
    for(auto A :a){
        for(int i = 2; i <= A; ){
            if(A % i == 0){
                set.insert(i);
                A = A / i;
                if(A == 1) {
                    continue;
                }
            }else {
                i++;
            }
            
        }
    }
    return set.size() <= n? "Yes":"No"; // placeholder return
}

int main() {
    vector<int> a1 = {1, 2, 3, 4};
    vector<int> a2 = {10, 12};
    vector<int> a3 = {6, 9, 15};

    cout << (solution(4, a1) == "Yes") << endl;
    cout << (solution(2, a2) == "No") << endl;
    cout << (solution(3, a3) == "Yes") << endl;
    return 0;
}