map映射巧用 A-B Problems
A-B problem
Description
大家都非常熟悉 A+B Problem! 题目看多了也有审美疲劳,于是我舍弃了,改用 A-B problem! 题目是这样的:给出一串数以及一个数字 C,要求计算出所有 A-B=C 的数对 的个数。( 注意: 不同位置的数字一样的数对算不同的数对)
Input Format
第一行包括 2 个非负整数 N 和 C,中间用空格隔开。 第二行有 N 个整数,中间用空格隔开,作为要求处理的那串数。
Output Format
输出一行,表示该串数中包含的所有满足 A-B=C 的数对的个数。
Sample Input
4 1
1 1 2 3
Sample Output
3
Data Limit
对于 50%的数据, N <= 2000; 对于 100%的数据, N <= 200000。
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
#define ll long long
using namespace std;
int main()
{
int n;
ll c;
scanf("%d%lld",&n,&c);
ll cnt[222222];
map<ll,int>mm;
for(int i=0;i<n;i++)
{
scanf("%lld",&cnt[i]);
mm[cnt[i]]++;
}
ll ans=0;
for(int i=0;i<n;i++)
{
if(c!=0)
ans+=mm[cnt[i]+c];
if(c==0)
ans+=mm[cnt[i]+c]-1;
}
printf("%lld\n",ans);
return 0;
}