Codeforces Round #708 (Div. 2) E2. Square-free division (hard version)

Codeforces Round #708 (Div. 2) E2. Square-free division (hard version)

题意

长度为n的数组，可以修改k次，然后切分成多个数组，每个数组满足任意两个数相乘不是一个平方数

题解

满足相乘不是平方，就需要对每个数做质因子分解，这块需要优化，否则会T，具体看代码理解吧

修改k次就需要dp，因为k最大20，所以可以维护一个相同数字下标的队列，从大到小，最多21个数，表示需要修改的数字，然后dp

有看博客说时间复杂度是 n*k*k 的，这会T，就看了下思路没有看代码，我自己写是T了，可能是写的挫吧。

代码

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <map>
using namespace std;
#define N 200005
#define M 10000005
const int INF = 1e9 + 7;
int prime[M], tot, pre[M];
bool ji[M];
int a[M], dp[N][21], left[N][21], b[N], c[25];
map<int, int> mp;
bool cmp(int x, int y)
{
    return x > y;
}
void init()
{
    for (int i = 2; i < M; i++)
    {
        if (!ji[i])
        {
            prime[tot++] = i;
            pre[i] = i;
        }
        for (int j = 0; j < tot && i * prime[j] < M; j++)
        {
            ji[i * prime[j]] = 1;
            pre[i * prime[j]] = prime[j];
            if (i % prime[j] == 0)
                break;
        }
    }
}
int main()
{
    init();
    int t, n, k, m;
    for (scanf("%d", &t); t; t--)
    {
        scanf("%d%d", &n, &k);
        mp.clear();
        int cnt = 0;
        c[++cnt] = 0;
        for (int i = 1; i <= n; i++)
        {
            m = 0;
            scanf("%d", &a[i]);
            for (int j = a[i]; j > 1; j /= pre[j])
            {
                if (m && b[m] == pre[j])
                    m--;
                else
                    b[++m] = pre[j];
            }
            for (a[i] = 1; m; m--)
                a[i] *= b[m];
            if (mp[a[i]])
                c[++cnt] = mp[a[i]];
            mp[a[i]] = i;
            sort(c + 1, c + cnt + 1, cmp);
            cnt = min(cnt, 21);
            for (int j = 0; j <= k; j++)
            {
                dp[i][j] = n + 1;
                for (int x = 0; x < cnt && x <= j; x++)
                    dp[i][j] = min(dp[i][j], dp[c[x + 1]][j - x] + 1);
            }
        }
        int ans = n + 1;
        for (int i = 0; i <= k; i++)
            ans = min(ans, dp[n][i]);
        printf("%d\n", ans);
    }

    return 0;
}