当前位置: 代码迷 >> 综合 >> LeetCode刷题:279. Perfect Squares 完美平方数的DP解法
  详细解决方案

LeetCode刷题:279. Perfect Squares 完美平方数的DP解法

热度:86   发布时间:2024-01-15 19:43:32.0

LeetCode刷题:279. Perfect Squares 完美平方数的DP解法

原题链接:https://leetcode.com/problems/perfect-squares/

Given a positive integer n, find the least number of perfect square numbers (for example, 1, 4, 9, 16, ...) which sum to n.

Example 1:

Input: n = 12
Output: 3 
Explanation: 12 = 4 + 4 + 4.
Example 2:

Input: n = 13
Output: 2
Explanation: 13 = 4 + 9.

问题描述:给定一个正整数N,这个正整数N可以用最少的整数的平方和表示。 
例如:12可以表示为 4+4+4,返回值 n=3,即 3个4之和。而4是2的平方。 
给定整数13,13可以表示为 4+9, 返回值 n=2,即 2的平方和加上3的平方和。

问题分析:这个题目实际上描述的是的Lagrange四平方定理。更加确切的说法应该是:给定一个正整数N,这个正整数N可以用不超过4个整数的平方和表示。 这个定理在数学上已经被证明是正确的。我们不用证明,只需要使用这个定理来设计算法即可。

Lagrange 四平方定理: 任何一个正整数都可以表示成不超过四个整数的平方之和。

那么我们这个问题的解法就变得很简单了,我们的结果只有1,2,3,4,四种可能。

算法分析:采用DP来求解。

我们先看一个分析表格:

n的取值从1到20.

黄色格子分别表示1,2,3,4的平方值,结果为:1,4,9,16

而我们通过表格中的数值,可以找到一个规律:红色部分表示平方数,所有的完美平方数都可以看做一个普通数加上一个完美平方数,那么采用DP求解时的状态方程就是:

dp[i + j * j] = Math.min(dp[i] + 1, dp[i + j * j])

算法设计:

package com.bean.algorithm.dp;import java.util.Arrays;public class PerfectSquares2 {public static int numSquares(int n) {//开辟一个dp数组int[] dp = new int[n+1];//预设值Arrays.fill(dp, Integer.MAX_VALUE);for(int i = 0; i * i <= n; i++) dp[i * i] = 1;for(int i = 1; i <= n; i++) {  for(int j = 1; i + j * j <= n; j++) {  dp[i + j * j] = Math.min(dp[i] + 1, dp[i + j * j]);  }}return dp[n];}public static void main(String[] args) {// TODO Auto-generated method stubint number=12;int result = numSquares(number);System.out.println("result = "+result);}}

输出结果:

result = 3