黎明者 pat basic 练习八十三 是否存在相等的差

题目:

给定 N 张卡片，正面分别写上 1、2、……、N，然后全部翻面，洗牌，在背面分别写上 1、2、……、N。将每张牌的正反两面数字相减（大减小），得到 N 个非负差值，其中是否存在相等的差？

输入格式：

输入第一行给出一个正整数 N（2 ≤ N ≤ 10 000），随后一行给出 1 到 N 的一个洗牌后的排列，第 i 个数表示正面写了 i 的那张卡片背面的数字。

输出格式：

按照“差值 重复次数”的格式从大到小输出重复的差值及其重复的次数，每行输出一个结果。

输入样例：

8
3 5 8 6 2 1 4 7

 

输出样例：

5 2
3 3
2 2

 

 

 

思路:

 使用map，差作为key，重复次数作为value，然后输出value大于1的就好，不过题目要求按照key降序输出，所以还要重载operator()来实现降序

 

代码:

#include<iostream>
#include<map>
#include<cmath>
using namespace std;
struct cmp
{
    bool operator()(int a,int b)
    {
        return a>b;
    }
};
int main()
{
    int num;
    cin>>num;
    map<int,int,cmp> m;
    for(int i=0;i<num;i++)
    {
        int temp,diff;
        cin>>temp;
        diff=abs(temp-i-1);
        m[diff]++;
    }
    for(auto it=m.begin();it!=m.end();it++)
    {
        if(it->second>1)
            cout<<it->first<<" "<<it->second<<endl;
    }
    return 0;
}