洛谷P1102

A-B 数对 [P1102]

题目描述

出题是一件痛苦的事情！

相同的题目看多了也会有审美疲劳，于是我舍弃了大家所熟悉的 A+B Problem，改用 A-B 了哈哈！

好吧，题目是这样的：给出一串数以及一个数字 \(C\)，要求计算出所有 \(A - B = C\) 的数对的个数（不同位置的数字一样的数对算不同的数对）。

输入格式

输入共两行。

第一行，两个整数 \(N, C\)。

第二行，\(N\) 个整数，作为要求处理的那串数。

输出格式

一行，表示该串数中包含的满足 \(A - B = C\) 的数对的个数。

样例 #1

样例输入 #1

4 1
1 1 2 3

样例输出 #1

3

提示

对于 \(75\%\) 的数据，\(1 \leq N \leq 2000\)。

对于 \(100\%\) 的数据，\(1 \leq N \leq 2 \times 10^5\)。

保证所有输入数据绝对值小于 \(2^{30}\)，且 \(C \ge 1\)。

2017/4/29 新添数据两组

题解

76pts解法

#include<bits/stdc++.h>
using namespace std;
int main(){
    long long cnt=0;
    long long n,c;
    long long arr[30000+1];
    cin>>n>>c;
    for(int i=0;i<n;i++){
        cin>>arr[i];
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            if(arr[j]-arr[i]==c)cnt++;
        }
    }
    cout<<cnt<<endl;
    return 0;
}

[TLE]*3

map解法

针对数组arr作映射，映射到每一个数字的出现次数；然后整个arr数组减去c；

这样直接统计map中和arr数组一样的就行了

#include<bits/stdc++.h>
using namespace std;
int main(){
    long long count=0;
    long long n,c;
    long long arr[30000+1];
       map<long long,long long> cnt;
    cin>>n>>c;
    for(long long i=0;i<n;i++){
        cin>>arr[i];
        cnt[arr[i]]++;
        arr[i]-=c;
    }
    for(long long i=0;i<n;i++){
            count+=cnt[arr[i]];
    }
    cout<<count<<endl;
    return 0;
}

关于我改成scanf和printf就过了这件事

#include<iostream>
#include<cstdio> 
#include<cmath>
#include<map>
#include<algorithm>
using namespace std;
int n,c;
long long ans;
map<int,int> a;
int num[200005];
int main()
{
    scanf("%d%d",&n,&c);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&num[i]);
        a[num[i]]++;    //当前数的个数++ 
    }
    for(int i=1;i<=n;i++)
    {
        ans+=a[num[i]+c];    //答案+=相差为c的数的个数，即a[num[i]+c]位置的数的个数 
    }
    printf("%lld",ans);
    return 0;
}