给你一个下标从 0 开始的整数数组 nums
。如果下标对 i
、j
满足 0 ≤ i < j < nums.length
,如果 nums[i]
的 第一个数字 和 nums[j]
的 最后一个数字 互质 ,则认为 nums[i]
和 nums[j]
是一组 美丽下标对 。
返回 nums
中 美丽下标对 的总数目。
对于两个整数 x
和 y
,如果不存在大于 1 的整数可以整除它们,则认为 x
和 y
互质 。换而言之,如果 gcd(x, y) == 1
,则认为 x
和 y
互质,其中 gcd(x, y)
是 x
和 y
的 最大公因数 。
示例 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<j且num[i]第一个数字和nums[j]最后一个数字互质的i和j的对数。
分析:
(1)由于下标小的数是取第一个数字,因此可以在遍历数组的过程中,存储已经遍历过的数字的第一个数字,加速后面数字与前面数字的互质判断;
(2)由于判断的两个数都是个位数,因此可用数组count[k]表示已遍历过的数字中k的个数,则可以在遍历数组的过程中,枚举1-9,若nums[i]和数字k互质,则美丽下标对ans+=count[k]。
class Solution {
public:
int countBeautifulPairs(vector<int>& nums) {
int ans=0,num,count[10]={};
for(int x:nums){
num=x%10;
for(int i=1;i<10;++i){
if(gcd(i,num)==1) ans+=count[i];
}
while(x>=10) x/=10;
++count[x];
}
return ans;
}
};