leetcode-279-完全平方数-java

题目及测试

package pid279;
/*  Perfect Squares

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, ...）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例 1:

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

示例 2:

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


*/
public class main {
	
	public static void main(String[] args) {
		int[] testTable = {12,13,25,34};
		for (int i=0;i<testTable.length;i++) {
			test(testTable[i]);
		}
	}
		 
	private static void test(int ito) {
		Solution solution = new Solution();
		int rtn;
		long begin = System.currentTimeMillis();
		    System.out.print(ito+" ");
		    System.out.println();
		rtn = solution.numSquares(ito);//执行程序
		long end = System.currentTimeMillis();		
		    System.out.print(rtn+" ");
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功，353ms，极慢）

dp[i] = MIN(dp[i], dp[i - j] + 1)

因为遍历了之前所有的情况，所以速度很慢

package pid279;


public class Solution {
	public int numSquares(int n) {
        if(n==1){
        	return 1;
        }
        int[] dp=new int[n+1];
        dp[0]=Integer.MAX_VALUE;
        int i=1;
        while(i*i<=n){
        	dp[i*i]=1;
        	i++;
        }
        if(dp[n]==1){
        	return 1;
        }
        for(i=2;i<=n;i++){
        	if(dp[i]==1){
        		continue;
        	}
        	int min=Integer.MAX_VALUE;
        	for(int j=1;j<=i/2+1;j++){
        		if(dp[j]+dp[i-j]<min){
        			min=dp[j]+dp[i-j];
        		}
        	}
        	dp[i]=min;
        }
		
		return dp[n];
    }

}

解法2（别人的）

首先初始化长度为n+1的数组dp，每个位置都为0
如果n为0，则结果为0
对数组进行遍历，下标为i，每次都将当前数字先更新为最大的结果，即dp[i]=i，比如i=4，最坏结果为4=1+1+1+1即为4个数字
动态转移方程为：dp[i] = MIN(dp[i], dp[i - j * j] + 1)，i表示当前数字，j*j表示平方数
时间复杂度：O(n*sqrt(n))，sqrt为平方根

class Solution {
    public int numSquares(int n) {
        int[] dp = new int[n + 1]; // 默认初始化值都为0
        for (int i = 1; i <= n; i++) {
            dp[i] = i; // 最坏的情况就是每次+1
            for (int j = 1; i - j * j >= 0; j++) { 
                dp[i] = Math.min(dp[i], dp[i - j * j] + 1); // 动态转移方程
            }
        }
        return dp[n];
    }
}