Codeforces Round #596 (Div. 2, based on Technocup 2020 Elimination Round 2)

比赛链接：Codeforces Round #596 (Div. 2, based on Technocup 2020 Elimination Round 2)

官方题解：Technocup 2020 — Elimination Round 2 + Codeforces Round 596: analysis

A. Forgetting Things

题意

有两个数 (a) 和 (b)，给定 (a) 的首位 (d_a) 和 (b) 的首位 (d_b)，问能否构造出 (a) 和 (b)，满足 (a + 1 = b)

思路

如果 (a = b)，构造 (a1)，(b2)。

如果 (a + 1 = b)，构造 (a9)，(b0)。

如果 (a = 9)，(b = 1)，构造 (9)，(10)。

否则无法构造。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
    int n, m;
    cin >> n >> m;
    if(n == m) {
        printf("%d1 %d2
", n, m);
    } else if(n == 9 && m == 1) {
        printf("9 10
");
    } else if(n + 1 == m) {
        printf("%d9 %d0
", n, m);
    } else {
        printf("-1
");
    }
    return 0;
}

B1. TV Subscriptions (Easy Version)

题意

某电视节目有 (k) 集，告诉你 (n) 天每天播放哪一集，现在要看连续 (d) 天，最少需要买几集。

思路

暴力。

代码

#include <bits/stdc++.h>
using namespace std;
 
int a[110];
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    cin >> T;
    while(T--) {
        int n, k, d;
        cin >> n >> k >> d;
        for(int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        int ans = 0x3f3f3f3f;
        map<int, int> mp;
        for(int i = 0; i < n - d + 1; ++i) {
            mp.clear();
            for(int j = 0; j < d; ++j) {
                mp[a[i + j]]++;  
            }
            int tmp = mp.size();
            ans = min(ans, tmp);
        }
        int tmp = mp.size();
        ans = min(ans, tmp);
        cout << ans << endl;
    }
    return 0;
}

B2. TV Subscriptions (Hard Version)

题意

与上一题相同，数据范围变大。

思路

维护一个长度为 (d) 的滑动窗口即可。

代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
 
int a[maxn];
int mp[maxn * 5];
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    cin >> T;
    while(T--) {
        memset(mp, 0, sizeof(mp));
        int n, k, d;
        cin >> n >> k >> d;
        for(int i = 0; i < n; ++i) {
            cin >> a[i];
        }
        int ans = 0x3f3f3f3f;
        int tmp = 0;
        for(int i = 0; i < n; ++i) {
            if(i < d) {
                if(mp[a[i]] == 0) ++tmp;
                mp[a[i]]++;
            } else {
                ans = min(ans, tmp);
                mp[a[i - d]]--;
                if(mp[a[i - d]] == 0) --tmp;
                if(mp[a[i]] == 0) ++tmp;
                mp[a[i]]++;
            }
        }
        ans = min(ans, tmp);
        cout << ans << endl;
    }
    return 0;
}

C. p-binary

题意

给定两个整数 (n) 和 (p)，将 (n) 分解成若干个 (2^x + p) 的和，求最少需要几项。如 (n=24,p=-1)，则 (24 = (2^4 - 1) + (2^2 - 1) + (2^2 - 1) + (2^2 - 1))，所以答案为 (4)。

思路

枚举项数 (i)，(2^{31}>1e9)，所以不超过 (31) 项。然后将 (n) 减去 (p*i)，(n - p imes i) 最多有 (n - p imes i) 项 (每项都为 (2^0))，最少项数为二进制分解后二进制中为 (1) 的个数，判断 (i) 是否在这个范围内即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
    int n, p;
    int ans = -1, tmp = 0;
    scanf("%d%d", &n, &p);
    for(int i = 1; i <= 31; ++i) {
        int cnt = 0;
        tmp = n - p * i;
        while(tmp > 0) {
            if(tmp & 1) {
                cnt++;
            }
            tmp >>= 1;
        }
        if(cnt <= i && i <= n - p * i){
            ans = i;
            break;
        }
    }
    printf("%d
", ans);
    return 0;
}

D. Power Products

题意

给定 (n) 个正整数 (a_1, a_2, ..., a_n)，和一个整数 (kge 2)，问有多少对 (a_i, a_j) 满足 (a_icdot a_j = x^k)，(x) 是整数。

思路

将每个数质因数分解，将每个因子和模 (k) 后的数量存入 (map)，然后找所需的剩余因子个数满足的是否存在。

比如 (24 = 2^3 * 3)，设 (k = 2)，则 (map) 中存 (((2, 3mod k), (3, 1 mod k)))，(((2, 1), (3, 1)))。然后在 (map) 中找 (((2,k - 1), (3, k - 1))) 是否存在即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
 
ll a[maxn];
 
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll n, k;
    cin >> n >> k;
    map<vector<pair<ll, ll> >, ll> mp;
    ll ans = 0;
    for(int i = 0; i < n; ++i) {
        cin >> a[i];
        vector<pair<ll, ll> > vt;
        for(int j = 2; j <= a[i] / j; ++j) {
            int num = 0;
            if(a[i] % j == 0) {
                while(a[i] % j == 0) {
                    ++num;
                    a[i] /= j;
                }
            }
            if(num % k) {
                vt.push_back({j, num % k});
            }
        }
        if(a[i] > 1) vt.push_back({a[i], 1 % k});
        vector<pair<ll, ll> > vt2;
        for(int j = 0; j < vt.size(); ++j) {
            vt2.push_back({vt[j].first, k - vt[j].second});
        }
        ans += mp[vt2];
        mp[vt]++;
    }
    cout << ans << endl;
    return 0;
}