题目
通过的用户数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;
}