ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

5708. 统计一个数组中好对子的数目

2021-04-04 10:32:43  阅读:191  来源: 互联网

标签:11 5708 nums int rev ans 中好 对子 97


给你一个数组 nums ,数组中只包含非负整数。定义 rev(x) 的值为将整数 x 各个数字位反转得到的结果。比方说 rev(123) = 321 , rev(120) = 21 。我们称满足下面条件的下标对 (i, j) 是 好的 :

  • 0 <= i < j < nums.length
  • nums[i] + rev(nums[j]) == nums[j] + rev(nums[i])

请你返回好下标对的数目。由于结果可能会很大,请将结果对 109 + 7 取余 后返回。

 

示例 1:

输入:nums = [42,11,1,97]
输出:2
解释:两个坐标对为:
 - (0,3):42 + rev(97) = 42 + 79 = 121, 97 + rev(42) = 97 + 24 = 121 。
 - (1,2):11 + rev(1) = 11 + 1 = 12, 1 + rev(11) = 1 + 11 = 12 。

示例 2:

输入:nums = [13,10,35,24,76]
输出:4

 

提示:

  • 1 <= nums.length <= 105
  • 0 <= nums[i] <= 109

 

 

nums[i] + rev(nums[j]) == nums[j] + rev(nums[i])=>

(nums[i] - rev(nums[i])) == (nums[j] - rev(nums[j]))=>

Transform each nums[i] into (nums[i] - rev(nums[i])). Then, count the number of (i, j) pairs that have equal values.=>

add  c*(c-1)/2  to the ans for each c in the Counter()

 

python

class Solution:
    def countNicePairs(self, nums: List[int]) -> int:
        for i in range(len(nums)):
            nums[i]=nums[i]-int(str(nums[i])[::-1])
        cnt=collections.Counter(nums)
        ans=0
        for c in cnt.values():
            ans+=(c*(c-1)/2)
        return int(ans)%(10**9+7)

C++

class Solution {
public:
    int countNicePairs(vector<int>& nums) {
        long ans=0;
        unordered_map<int,int> m;
        for(int n:nums){
            string s=to_string(n);
            reverse(s.begin(),s.end());
            ans+=(m[n-stoi(s)])++;
        }
        return ans%(long)(1e9+7); 
    }
};

 

标签:11,5708,nums,int,rev,ans,中好,对子,97
来源: https://www.cnblogs.com/xxxsans/p/14615682.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有