題目鏈接

題意

給你N個數字和一個K，問一共有幾種拼接數字的方式使得到的數字是K的倍數，拼接：“234”和“123”拼接得到“234123”

AC

• N <= 2e5,簡單的暴力O（N^2）枚舉肯定超時
• 數字A和B拼接，B的位數最多10位，如果我們知道位數為（1-10）的數字和A拼接滿足是K的倍數這樣的數字有幾個，就可以在N*10的復雜度下完成所有的拼接
• 在讀入數據的時候，我們可以統計出數字的位數和對K取余的結果，這樣我們就可以在O(1)的時間內得到所有滿足的情況

#include <iostream> #include <cmath> #include <map> #include <vector> #include <set> #include <algorithm> #define ll long long #define ull unsigned long long #define N 200005 using namespace std;ll a[N]; // 獲取長度 int getlen(int x) {int sum = 0;while (x) {++sum;x /= 10;}return sum; } map<int, int> mp[11]; int main() { #ifndef ONLINE_JUDGEfreopen("in.txt", "r", stdin); #endif int n, k;while (scanf("%d%d", &n, &k) != EOF) {for (int i = 0; i < n; ++i) {scanf("%lld", &a[i]);int len = getlen(a[i]);mp[len][a[i] % k]++;}ll ans = 0;// 枚舉N個數字 for (int i = 0; i < n; ++i) {ll mul = 1;int len = getlen(a[i]);mp[len][a[i] % k]--;// 對每個數字，分別對位數為 j 的數字進行拼接 for (int j = 1; j <= 10; ++j) {mul *= 10;ull temp = a[i] * mul;if (mp[j].find((k - temp % k) % k) != mp[j].end())ans += mp[j][(k - temp % k) % k];}mp[len][a[i] % k]++;}printf("%lld\n", ans);for (int i = 0; i < 11; ++i)mp[i].clear();}return 0; }

總結

以上是生活随笔為你收集整理的Codeforces Round #506 (Div. 3) - D. Concatenated Multiples（思维）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。