I'm Prime

十方三世,尽在一念

View on GitHub

leetcode 笔记


题目描述

给你一个整数 n 请你判断 n 是否为 丑数 如果是返回 true 否则返回 false 

丑数 就是只包含质因数 23 / 5 的正整数

 

示例 1

输入n = 6
输出true
解释6 = 2 × 3
示例 2

输入n = 8
输出true
解释8 = 2 × 2 × 2
示例 3

输入n = 14
输出false
解释14 不是丑数因为它包含了另外一个质因数 7 
示例 4

输入n = 1
输出true
解释1 通常被视为丑数
 

提示

-231 <= n <= 231 - 1

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ugly-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

code

class Solution {
public:
    bool isUgly(int n) {
        if (n <= 0) return false;
        int arr[] = {2, 3, 5};
        for (int i : arr) while (n % i == 0) n /= i;
        return n == 1;
    }
};

一刀流,每次会有array(2, 3, 5)内存回收,和find{ n % it == 0 }的额外开销。

class Solution {
    tailrec fun isUgly(n: Int): Boolean = when {
        n <= 0 -> false
        n == 1 -> true
        else -> when (val i = arrayOf(2, 3, 5).find { n % it == 0 }) {
            null -> false
            else -> isUgly(n / i)
        }
    }
}

常规:

class Solution {
    fun isUgly(n: Int): Boolean = when {
        n <= 0 -> false
        n == 1 -> true
        else -> arrayOf(2, 3, 5).fold(n){ acc, i -> 
            var r = acc
            while (r % i == 0) r /= i
            r
        } == 1
    }
}
object Solution {
  def isUgly(n: Int): Boolean = n match {
    case 0 => false
    case 1 => true
    case _ => Seq(2, 3, 5).foldLeft(n) { (acc, i) =>
      var r = acc
      while (r % i == 0) r /= i
      r
    } == 1
  }
}

rust 一刀流正合适

vec![2, 3, 5].iter().fold(n, |mut acc, i|{ while acc % i == 0 {acc /= i}; acc}) == 1
impl Solution {
    pub fn is_ugly(mut n: i32) -> bool {
        match n {
            0 => false,
            1 => true,
            _ => {
                vec![2, 3, 5].iter().fold(n, |mut acc, i| {
                    while acc % i == 0 {
                        acc /= i
                    }
                    acc
                }) == 1
            }
        }
    }
}