统计坏数对的数目

题目

通过的用户数1634
尝试过的用户数3003
用户总通过次数1660
用户总提交次数5338
题目难度Medium
给你一个下标从 0 开始的整数数组 nums 。如果 i < j 且 j - i != nums[j] - nums[i] ,那么我们称 (i, j) 是一个 坏数对 。

请你返回 nums 中 坏数对 的总数目。

示例 1:

输入:nums = [4,1,3,3]
输出:5
解释:数对 (0, 1) 是坏数对,因为 1 - 0 != 1 - 4 。
数对 (0, 2) 是坏数对,因为 2 - 0 != 3 - 4, 2 != -1 。
数对 (0, 3) 是坏数对,因为 3 - 0 != 3 - 4, 3 != -1 。
数对 (1, 2) 是坏数对,因为 2 - 1 != 3 - 1, 1 != 2 。
数对 (2, 3) 是坏数对,因为 3 - 2 != 3 - 3, 1 != 0 。
总共有 5 个坏数对,所以我们返回 5 。
示例 2:

输入:nums = [1,2,3,4,5]
输出:0
解释:没有坏数对。

提示:

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

收获

1:看到此题万不可直接不动脑就进行双层循环遍历,因为在最后会超出时间限制,所以需要进行一定的变行,这里进行学习
2:坏数对就是i <j 且 j - i != nums [j] - nums [i],这里进行变形nums[j]-j!=nums[i]-i;(i<j),正向思维找所有的nums[i]的话会超时,所以这里逆向思维找nums[i]-i=nums[j]-j的对数,创建map映射,首先找到每一个位置第一个的mao[i]-i的值,接下来如果有数相同的话,对应的数目就++,用所有可能的个数n*(n-1)/2减去最后得到的映射对数,就可以得到坏对数的数目了~
3:在创建unordered_map时直接用#include <bits/stdc++>会报错,上网查询好像是编译器版本过低的问题,没法直接使用,就需要加上头文件#include<tr1/unordered_map>和using namespace std::tr1;才能进行编译~
4:1LL的问题参考博文

代码

#include <iostream>
#include <bits/stdc++.h>
#include<tr1/unordered_map>

using namespace std::tr1;
using namespace std;

long long conutBadPairs(vector<int>& nums){
    int n=nums.size();
    long long ans=1LL*n*(n-1)/2;
    unordered_map<int, int> mp;
    for(int i=0;i<n;i++){
        int t=nums[i]-i;
        ans-=mp[t];
        //这里的++不能放在前面要放在后面等到下一次有相同映射的时候再去真正的-
        mp[t]++;
    }
    return ans;
}

int main()
{
    vector<int> nums;
    int temp;
    while(cin>>temp){
        nums.push_back(temp);
        if(cin.get()=='\n'){
            break;
        }
    }
    cout<<conutBadPairs(nums);
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值