2020第十一届蓝桥杯7月省赛I题
试题 I: 整数拼接
题目描述:
本题思路:
这道题,暴力过30%的样例,拿的了7分的样子,要全过的话那得dp,写数学式子分析,找到尽可能减少复杂度的方法。
先上暴力代码(这道题暴力写应该很容易的):
#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
typedef long long ll;
int n, k;
int a[N];
int main()
{
scanf ("%d%d", &n, &k);
int cnt = 0;
for (int i = 0; i < n; i++)
{
scanf ("%d", &a[i]);
}
for (int i = 0; i < n - 1; i++)
{
for (int j = i + 1; j < n; j++)
{
int x = a[i];
int y = a[j];
if (!(x % k) && !(y % k))
{
cnt += 2;
continue;
}
int lenx = log10(x) + 1;
int leny = log10(y) + 1;
ll nowx = y * pow(10, lenx) + x;
ll nowy = x * pow(10, leny) + y;
if (!(nowx % k))
{
cnt++;
}
if (!(nowy % k))
{
cnt++;
}
}
}
cout << cnt << endl;
return 0;
}
数学分析:
判断两个数的拼接是否能被k整除,假设当前操作数为a[i], a[j]。
拼接之后的数就是a[i]*10^(log10a[j]) + a[j],对k取模,式子就是(a[i]*10 ^ (log10a[j] + 1) % k + a[j] % k)% k,这个式子的范围在0~10 ^ 5之间,我们对这个式子进行分解分析:
第一部分:a[ i ] * 10 ^ (log10a[j] + 1) % k,其实就可以算出a[i]乘以不同的幂对k取模值为前面那个式子的个数,即用dp[幂][a[i]*10^(log10a[j]) % k]记录个数
第二部分:a[j] % k,就可以根据上面得到的个数直接找到k - a[j] % k有多少个,这两项之和可以被k整除,因为a[j] % k 可能为0,所以就再取一次模,(k - a[j] % k)% k.
然后在遍历的时候直接累加到ans里面就行
代码部分solve()函数解决的就是记录个数的部分,但是因为是在计数部分进行ans的更新的,所以ans记录的只有当前某个操作数a[i]之前拼接的个数。
所以我们还得反向进行一次计算,即直接把数组a[]翻转一下,mst一下cnt[][],再跑一遍solve()即可。
最后输出ans就可以啦~
代码部分:
#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int a[N];
int cnt[11][N];//cnt[i][j]表示i次幂下模k后值为j的个数
int n, k;
ll ans;
void solve()
{
for (int i = 0; i < n; i++)
{
ll t = 1;
ans += cnt[(int)log10(a[i]) + 1][(k - a[i] % k) % k];
for (int j = 1; j < 11; j++)
{
t = pow(10, j);
cnt[j][a[i] % k * t % k]++;
}
}
}
int main()
{
scanf ("%d%d", &n, &k);
for (int i = 0; i < n; i++)
{
scanf ("%d", &a[i]);
}
solve();//拼接a[j][i]
mst(cnt, 0);
reverse(a, a + n);
solve();//拼接a[i]a[j]
cout << ans << endl;
return 0;
}