组合数类型结合模的应用 && YTUOJ 4294 && Codeforces Round 943 (Div. 3) D 题

前序：
 

首先，我先对于我这篇文章所起的题目进行解释，为什么这篇文章叫组合数结合模的应用，今天我来讲解引进一类题型，这类题型呢涉及模 mod 的应用，然后呢，最终类似于组合数Cn2, 就是有n种是符合的，我们用这n种当中选出2种，结果是（n*(n-1)/2）其实也就是 0+1+2+3+......n-1 ，因此这类题型一般是结合map来实现的

先来一道例题：
 

YOJ 4294: cv工程师的整数2

题目描述

cv工程师 喜欢整数200 。请为他解答下面的问题。
给定一个由 N 个正整数组成的序列 A ，找出满足以下所有条件的整数对 (i,j)：

1≤i<j≤N1  ；
Ai−Aj 是200 的倍数。

输入中的所有值都是整数。
2≤N≤2×105
1≤Ai≤109

输入

输入格式如下：
N
A1​ A2 AN

输出

一个整数表示答案。

输入输出样例

样例输入 #1

复制

6
123 223 123 523 200 2000

样例输出 #1

复制

4

提示

样例解释：

对于 (i,j)=(1,3)， A1−A3=0是 200 的倍数。
我们有四对满足条件： (i,j)=(1,3),(1,4),(3,4),(5,6）

 

题干大意：

这道题说的是要求（A[i]-A[j]）mod 200=0 ， 首先根据模的相关性质，我们可以简单知道，A[i]%200=A[j]%200, 这样再结合前面前序所说的内容，结合map，就可以轻松解决，记得开long long 噢。

代码实现：

#include<bits/stdc++.h>

using i64=long long;

int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int n;
    std::cin>>n;

    std::map<int,int> cnt;

    i64 ans=0;
    for(int i=0;i<n;i++)
    {
        int a; std::cin>>a;
        a%=200;
        ans+=cnt[a];
        cnt[a]++;
    }

    std::cout<<ans<<"\n";
    return 0;
}

D. Divisible Pairs

原题链接:
https://codeforces.com/group/liw76rpnp5/contest/540021/problem/D

波利卡普有两个最喜欢的整数 x 和 y (它们可以相等)，他找到了一个长度为 n 的数组 a 。

波利卡普考虑了一对索引 〈i,j〉 (1≤i<j≤n ）是美丽的，如果：

• ai+aj 可以被 x 整除；
• ai−aj 可以被 y 整除。

例如，如果 x=5 ， y=2 ， n=6 ， a= [ 1,2,7,4,9,6 ]，那么唯一优美的一对是：

• 〈1,5〉 : a1+a5=1+9=10 ( 10 能被 5 整除)和 a1−a5=1−9=−8( −8 能被 2 整除)；
• 〈4,6〉 : a4+a6=4+6=10 ( 10 能被 55 整除)和a4−a6=4−6=−2 ( −2 能被 2 整除)。

求数组 a 中优美数对的个数。

思路：
 

这道题就是上一道的升级版，第一道题只有一个限制条件，而第二道题在第一道题的限制条件的基础上用加了 （a[i] +a[j] ）mod x= 0 ,这意味着 (a[i] mod x+ a[j] mod x) mod x=0; 所以我们要使用map 加 pair, pair 来记录要同时满足条件1 和 条件2 ，首先对于 条件：
(a[i] - a[j]) mod y =0; 在第一题当中已经结束过了，不再赘述， （a[i] +a[j] ）mod x= 0 ，对于一个输入a, 于其配对的是 (x-a mod x) mod x . 

代码实现：

#include<bits/stdc++.h>

using i64=long long;

void solve()
{
    int n,x,y;
    std::cin>>n>>x>>y;
    std::map<std::pair<int,int>,int> cnt;
    i64 ans=0;
    for(int i=0;i<n;i++)
    {
        int a;
        std::cin>>a;
        ans+=cnt[{(x-a%x)%x,a%y}];// 与输入的a配对，符合要求
        cnt[{a%x,a%y}]++;
    }
    std::cout<<ans<<"\n";
}

int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int t;
    std::cin>>t;

    while(t--)
    {
        solve();
    }
    return 0;
}