整数拼接(枚举，哈希表，数学)

题目链接:https://www.acwing.com/problem/content/2070/

题目
给定一个长度为 $n$ 的数组 $A_1,A_2,\cdot \cdot \cdot ,A_n$。

你可以从中选出两个数 $A_i$ 和 $A_j$(i 不等于 j)，然后将 $A_i$ 和 $A_j$ 一前一后拼成一个新的整数。

例如 $12$ 和 $345$ 可以拼成 $12345$ 或 $34512$。

注意交换 $A_i$ 和 $A_j$ 的顺序总是被视为 2 种拼法，即便是 $A_i=A_j$ 时。

请你计算有多少种拼法满足拼出的整数是 $K$ 的倍数。

输入格式
第一行包含 $2$ 个整数 $n$ 和 $K$。

第二行包含 $n$ 个整数 $A_1,A_2,\cdot \cdot \cdot ,A_n$。

输出格式
一个整数代表答案。

数据范围
$1\le n\le 10^5,$
$1\le K\le 10^5,$
$1\le A_i\le 10^9$

输入样例：

4 2
1 2 3 4

输出样例：

6

思路:

$A_j{A}_i$拼接起来的数可以表示为 $A_j^{len(A_i)}+A_i$
因为是$K$的倍数，所以有 $A_j^{len(A_i)}+A_i=0(modk)$
即 $A_j^{len(A_i)}=-A_i(modk)$
可以用十个哈希表存储 $A_j\ast 10^t，t\in [1,10]$ 模 $k$ 的值
接着只需要枚举每个 $A_i$，看有多少个 $A_j^{len(A_i)}\%k=-A_i\%k$ 即可
注意:
①因为负数取模，c++中得到的是负数，数学上定义的是正数
比如
$-5\%2=-3\ast 2+1$ 因此 $-5\%2=1$
$-1\%4=-1\ast 4+3$ 因此 $-1\%4=3$
要取得 $-a\%k$ 的正数，即 $(k-a)\%k$

②$A_j$和$A_i$是同一个数的情况，即 $j=i$，答案要减1

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

typedef long long LL;

const int N = 100010;

LL n,k,ans;
int arr[N],s[11][N];

int main()
{
    scanf("%lld%lld",&n,&k);
    for(int i = 0; i < n; i++)
    {
        scanf("%d",&arr[i]);
        LL t = arr[i] % k;
        t = t * 10 % k; // 从10^1开始
        for(int j = 1; j < 11; j++)
        {
            s[j][t]++;
            t = t * 10 % k;
        }
    }
    for(int i = 0; i < n; i++)
    {
        LL t = arr[i] % k;
        LL mod = (k - t) % k; //因为负数取模，c++中得到的是负数，要取得正数
        int len = to_string(arr[i]).size();
        ans += s[len][mod];
        LL r = t;
        while(len--) r = r * 10 % k;
        if(r == mod) ans --;
    }
    printf("%lld\n",ans);
    return 0;
}