码迷,mamicode.com
首页 > 其他好文 > 详细

279. Perfect Squares

时间:2020-07-17 01:15:12      阅读:72      评论:0      收藏:0      [点我收藏+]

标签:note   lin   class   array   while   tco   style   nta   evel   

package LeetCode_279

import java.util.*
import kotlin.collections.HashSet

/**
 * 279. Perfect Squares
 * https://leetcode.com/problems/perfect-squares/description/
 *
 * 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.
 */

class Solution {
    /*
    * solution:bfs, find out the shortest path be the answer
    * Time complexity:O(n*sqrt(n)), Space complexity:O(n)
    * */
    fun numSquares(n: Int): Int {
        if (n == 0) {
            return 0
        }
        val levels = IntArray(n + 1)
        val visited = HashSet<Int>()
        val queue = LinkedList<Int>()
        val start = 0
        visited.add(start)
        queue.offer(start)
        while (queue.isNotEmpty()) {
            val cur = queue.poll()
            if (cur == n) {
                return levels[cur]
            }
            var i = 1
            while (cur + i * i <= n) {
                val new = cur + i * i
                if (!visited.contains(new)) {
                    visited.add(new)
                    levels[new] = levels[cur] + 1
                    queue.offer(new)
                }
                i++
            }
        }
        return levels[start]
    }
}

 

279. Perfect Squares

标签:note   lin   class   array   while   tco   style   nta   evel   

原文地址:https://www.cnblogs.com/johnnyzhao/p/13326382.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!