LeetCode 2748.美丽下标对的数目:模拟

【LetMeFly】2748.美丽下标对的数目:模拟

力扣题目链接:https://leetcode.cn/problems/number-of-beautiful-pairs/

给你一个下标从 0 开始的整数数组 nums 。如果下标对 ij 满足 0 ≤ i < j < nums.length ,如果 nums[i]第一个数字nums[j]最后一个数字 互质 ,则认为 nums[i]nums[j] 是一组 美丽下标对

返回 nums美丽下标对 的总数目。

对于两个整数 xy ,如果不存在大于 1 的整数可以整除它们,则认为 xy 互质 。换而言之,如果 gcd(x, y) == 1 ,则认为 xy 互质,其中 gcd(x, y)xk 最大公因数

示例 1:

复制代码
输入:nums = [2,5,1,4]
输出:5
解释:nums 中共有 5 组美丽下标对:
i = 0 和 j = 1 :nums[0] 的第一个数字是 2 ,nums[1] 的最后一个数字是 5 。2 和 5 互质,因此 gcd(2,5) == 1 。
i = 0 和 j = 2 :nums[0] 的第一个数字是 2 ,nums[2] 的最后一个数字是 1 。2 和 5 互质,因此 gcd(2,1) == 1 。
i = 1 和 j = 2 :nums[1] 的第一个数字是 5 ,nums[2] 的最后一个数字是 1 。2 和 5 互质,因此 gcd(5,1) == 1 。
i = 1 和 j = 3 :nums[1] 的第一个数字是 5 ,nums[3] 的最后一个数字是 4 。2 和 5 互质,因此 gcd(5,4) == 1 。
i = 2 和 j = 3 :nums[2] 的第一个数字是 1 ,nums[3] 的最后一个数字是 4 。2 和 5 互质,因此 gcd(1,4) == 1 。
因此,返回 5 。

示例 2:

复制代码
输入:nums = [11,21,12]
输出:2
解释:共有 2 组美丽下标对:
i = 0 和 j = 1 :nums[0] 的第一个数字是 1 ,nums[1] 的最后一个数字是 1 。gcd(1,1) == 1 。
i = 0 和 j = 2 :nums[0] 的第一个数字是 1 ,nums[2] 的最后一个数字是 2 。gcd(1,2) == 1 。
因此,返回 2 。

提示:

  • 2 <= nums.length <= 100
  • 1 <= nums[i] <= 9999
  • nums[i] % 10 != 0

解题方法:模拟

两层循环模拟 i i i和 j j j即可。

获得某个数十进制下第一位的方法:

对于整数 n n n,当 n ≥ 10 n\geq 10 n≥10时不断令 n = ⌊ n 10 ⌋ n=\lfloor\frac{n}{10}\rfloor n=⌊10n⌋,最终的 n n n即为所求。

获得某个数十进制下最后一位的方法:

对于整数 n n n, n m o d    10 n\mod 10 nmod10即为所求。

获得两个数的最大公因数的方法:

可用辗转相除法。对于两个数 a a a和 b b b,令 c = a m o d    b c=a\mod b c=amodb:

  • 若 c = 0 c=0 c=0则 b b b即为所求;
  • 否则令 a , b = b , c a, b = b, c a,b=b,c。
  • 时间复杂度 O ( n 2 + n log ⁡ C ) O(n^2+n\log C) O(n2+nlogC),其中 C C C为一个数的最大值
  • 空间复杂度 O ( 1 ) O(1) O(1)

AC代码

C++
cpp 复制代码
class Solution {
private:
    inline int getFirst(int n) {
        while (n >= 10) {
            n /= 10;
        }
        return n;
    }
public:
    int countBeautifulPairs(vector<int>& nums) {
        int ans = 0;
        for (int i = 0; i < nums.size(); i++) {
            for (int j = i + 1; j < nums.size(); j++) {
                if (__gcd(getFirst(nums[i]), nums[j] % 10) == 1) {
                    ans++;
                }
            }
        }
        return ans;
    }
};
Go
go 复制代码
// package main

func getFirst(n int) int {
    for ; n >= 10; {
        n /= 10;
    }
    return n
}

func gcd(a int, b int) int {
    if b == 0 {
        return a
    }
    return gcd(b, a % b)
}

func countBeautifulPairs(nums []int) int {
    ans := 0
    for i := 0; i < len(nums); i++ {
        for j := i + 1; j < len(nums); j++ {
            if gcd(getFirst((nums[i])), nums[j] % 10) == 1 {
                ans++
            }
        }
    }
    return ans
}
Java
java 复制代码
// import java.math.BigInteger;

class Solution {
    private BigInteger getFirst(int n) {
        while (n >= 10) {
            n /= 10;
        }
        return BigInteger.valueOf(n);
    }

    public int countBeautifulPairs(int[] nums) {
        int ans = 0;
        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (getFirst(nums[i]).gcd(BigInteger.valueOf(nums[j] % 10)).equals(BigInteger.valueOf(1))) {
                    ans++;
                }
            }
        }
        return ans;
    }
}
Python
python 复制代码
from typing import List
from math import gcd

"""
[2,5,1,4]
"""
class Solution:
    def countBeautifulPairs(self, nums: List[int]) -> int:
        ans = 0
        for i in range(len(nums) - 1):
            for j in range(i + 1, len(nums)):
                if gcd(ord(str(nums[i])[0]) - ord('0'), nums[j] % 10) == 1:
                    ans += 1
        return ans

同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~

Tisfy:https://letmefly.blog.csdn.net/article/details/139842524

相关推荐
小刘|24 分钟前
《Java 实现希尔排序:原理剖析与代码详解》
java·算法·排序算法
jjyangyou29 分钟前
物联网核心安全系列——物联网安全需求
物联网·算法·安全·嵌入式·产品经理·硬件·产品设计
van叶~1 小时前
算法妙妙屋-------1.递归的深邃回响:二叉树的奇妙剪枝
c++·算法
简简单单做算法1 小时前
基于Retinex算法的图像去雾matlab仿真
算法·matlab·图像去雾·retinex
云卓SKYDROID1 小时前
除草机器人算法以及技术详解!
算法·机器人·科普·高科技·云卓科技·算法技术
半盏茶香1 小时前
【C语言】分支和循环详解(下)猜数字游戏
c语言·开发语言·c++·算法·游戏
徐子童1 小时前
双指针算法习题解答
算法
想要打 Acm 的小周同学呀2 小时前
LRU缓存算法
java·算法·缓存
劲夫学编程3 小时前
leetcode:杨辉三角
算法·leetcode·职场和发展
毕竟秋山澪3 小时前
孤岛的总面积(Dfs C#
算法·深度优先