题解 | kotori和素因子

解题思路

  • 对每个数分解质因数
  • 尝试每个数选择不同的质因数
  • 记录已使用的质因数,避免重复使用
  • 当选择完所有数时,更新最小和

代码

#include <iostream>
#include <vector>
#include <set>
using namespace std;

vector<int> getPrimeFactors(int n) {
    vector<int> factors;
    for(int i = 2; i * i <= n; i++) {
        if(n % i == 0) {
            while(n % i == 0) n /= i;
            factors.push_back(i);
        }
    }
    if(n > 1) factors.push_back(n);
    return factors;
}

int ans = 1e9;
void dfs(vector<vector<int>>& factors, vector<bool>& used, int pos, int sum) {
    if(pos == factors.size()) {
        ans = min(ans, sum);
        return;
    }
    
    for(int prime : factors[pos]) {
        if(!used[prime]) {
            used[prime] = true;
            dfs(factors, used, pos + 1, sum + prime);
            used[prime] = false;
        }
    }
}

int main() {
    int n;
    cin >> n;
    vector<int> nums(n);
    for(int i = 0; i < n; i++) cin >> nums[i];
    
    // 获取每个数的素因子
    vector<vector<int>> factors;
    for(int num : nums) {
        factors.push_back(getPrimeFactors(num));
    }
    
    vector<bool> used(1001, false);
    dfs(factors, used, 0, 0);
    
    cout << (ans == (int)1e9 ? -1 : ans) << endl;
    return 0;
}
import java.util.*;

public class Main {
    static int ans = Integer.MAX_VALUE;
    
    static List<Integer> getPrimeFactors(int n) {
        List<Integer> factors = new ArrayList<>();
        for(int i = 2; i * i <= n; i++) {
            if(n % i == 0) {
                while(n % i == 0) n /= i;
                factors.add(i);
            }
        }
        if(n > 1) factors.add(n);
        return factors;
    }
    
    static void dfs(List<List<Integer>> factors, boolean[] used, int pos, int sum) {
        if(pos == factors.size()) {
            ans = Math.min(ans, sum);
            return;
        }
        
        for(int prime : factors.get(pos)) {
            if(!used[prime]) {
                used[prime] = true;
                dfs(factors, used, pos + 1, sum + prime);
                used[prime] = false;
            }
        }
    }
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] nums = new int[n];
        for(int i = 0; i < n; i++) nums[i] = sc.nextInt();
        
        // 获取每个数的素因子
        List<List<Integer>> factors = new ArrayList<>();
        for(int num : nums) {
            factors.add(getPrimeFactors(num));
        }
        
        boolean[] used = new boolean[1001];
        dfs(factors, used, 0, 0);
        
        System.out.println(ans == Integer.MAX_VALUE ? -1 : ans);
    }
}

def get_prime_factors(n):
    factors = []
    i = 2
    while i * i <= n:
        if n % i == 0:
            while n % i == 0:
                n //= i
            factors.append(i)
        i += 1
    if n > 1:
        factors.append(n)
    return factors

def dfs(factors, used, pos, sum_val):
    global ans
    if pos == len(factors):
        ans = min(ans, sum_val)
        return
    
    for prime in factors[pos]:
        if not used[prime]:
            used[prime] = True
            dfs(factors, used, pos + 1, sum_val + prime)
            used[prime] = False

# 读入数据
n = int(input())
nums = list(map(int, input().split()))

# 获取每个数的素因子
factors = []
for num in nums:
    factors.append(get_prime_factors(num))

# DFS求解
ans = float('inf')
used = [False] * 1001
dfs(factors, used, 0, 0)

print(-1 if ans == float('inf') else ans)

算法及复杂度分析

  • 算法:
  • 时间复杂度:,其中 是数组长度, 是每个数的质因数个数
  • 空间复杂度:,用于存储质因数和标记数组
全部评论

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务