279. 完全平方数——(LeetCode刷题)

柔光的暖阳◎ 2024-04-01 19:46 126阅读 0赞

279. 完全平方数

给你一个整数 n ,返回 和为 n 的完全平方数的最少数量 。

完全平方数 是一个整数,其值等于另一个整数的平方;换句话说,其值等于一个整数自乘的积。例如,1、4、9 和 16 都是完全平方数,而 3 和 11 不是。

示例 1:

输入:n = 12
输出:3
解释:12 = 4 + 4 + 4

示例 2:

输入:n = 13
输出:2
解释:13 = 4 + 9

提示:

1 < = n < = 1 0 4 1 <= n <= 10^4 1<=n<=104

思路:

法一(BFS)

  • 可以将每个整数看成图中的一个节点,如果两个整数之差为一个平方数,那么这两个整数所在的节点就有一条边。
  • 要求解最小的平方数数量,就是求解从节点 n 到节点 0 的最短路径。
  • 本题也可以用动态规划求解,在之后动态规划部分中会再次出现。

法二:动态规划

可以将此题理解为完全背包问题。

  • 状态转移方程为:
    d p [ j ] = min ⁡ ( d p [ j ] , d p [ j − s q u a r e ] + 1 ) dp[j]=\min (dp[j], dp[j-square]+1) dp[j]=min(dp[j],dp[j−square]+1)

代码:(Java)

法一(BFS)

  1. import java.util.ArrayList;
  2. import java.util.LinkedList;
  3. import java.util.List;
  4. import java.util.Queue;
  5. public class bfs_per_squar {
  6. public static void main(String[] args) {
  7. // TODO 自动生成的方法存根
  8. int n = 12;
  9. System.out.println(numSquares(n));
  10. }
  11. public static int numSquares(int n) {
  12. List<Integer> squares = generateSquares(n);
  13. Queue<Integer> queue = new LinkedList<>();
  14. boolean[] marked = new boolean[n + 1];
  15. queue.add(n);
  16. marked[n] = true;
  17. int level = 0;
  18. while (!queue.isEmpty()) {
  19. int size = queue.size();
  20. level++;
  21. while(size-- > 0) {
  22. int cur = queue.poll();
  23. for(int s : squares) {
  24. int next = cur - s;
  25. if(next < 0) {
  26. break;
  27. }
  28. if(next == 0) {
  29. return level;
  30. }
  31. if(marked[next]) {
  32. continue;
  33. }
  34. marked[next] = true;
  35. queue.add(next);
  36. }
  37. }
  38. }
  39. return n;
  40. }
  41. private static List<Integer> generateSquares(int n) {
  42. // TODO 自动生成的方法存根
  43. List<Integer> squares = new ArrayList<>();
  44. for(int i = 1; i <= (int)Math.sqrt(n); i++) {
  45. squares.add(i*i);
  46. }
  47. return squares;
  48. }
  49. }

法二:动态规划

  1. import java.util.ArrayList;
  2. import java.util.List;
  3. public class PerfectSquares {
  4. public static void main(String[] args) {
  5. // TODO Auto-generated method stub
  6. int n = 12;
  7. System.out.println(numSquares(n));
  8. }
  9. public static int numSquares(int n) {
  10. List<Integer> squares = generateSquares(n);
  11. int []dp = new int[n + 1];
  12. for(int j = 1; j <= n; j++) {
  13. int min = Integer.MAX_VALUE;
  14. for(int square : squares) {
  15. if(square > j) break;
  16. min = Math.min(min, dp[j - square] + 1);
  17. }
  18. dp[j] = min;
  19. }
  20. return dp[n];
  21. }
  22. public static List<Integer> generateSquares(int n){
  23. List<Integer> squares = new ArrayList<>();
  24. for(int i = 1; i <= (int)Math.sqrt(n); i++) {
  25. squares.add(i * i);
  26. }
  27. return squares;
  28. }
  29. }

输出:

在这里插入图片描述

复杂度分析:(法二)

时间复杂度:共有 n ∗ n n ∗ \sqrt{n} n∗n 个状态需要转移,复杂度为 n ∗ n n * \sqrt{n} n∗n。
空间复杂度:O(n)。

注:仅供学习参考

来源:力扣

发表评论

表情:
评论列表 (有 0 条评论,126人围观)

还没有评论,来说两句吧...

相关阅读

    相关 leetcode279 完全平方

    给定正整数 n,找到若干个完全平方数(比如 1, 4, 9, 16, ...)使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。 示例 1: 输入: n = 12