CF1497E1 Square-free division (easy version)(set写法)

Square-free division (easy version)

This is the easy version of the problem. The only difference is that in this version k=0.

 

There is an array a1,a2,…,an of n positive integers. You should divide it into a minimal number of continuous segments, such that in each segment there are no two numbers (on different positions), whose product is a perfect square.

 

Moreover, it is allowed to do at most k such operations before the division: choose a number in the array and change its value to any positive integer. But in this version k=0, so it is not important.

 

What is the minimum number of continuous segments you should use if you will make changes optimally?

Input

The first line contains a single integer t (1≤t≤1000)  — the number of test cases.

 

The first line of each test case contains two integers n, k (1≤n≤2⋅105, k=0).

 

The second line of each test case contains n integers a1,a2,…,an (1≤ai≤107).

 

It's guaranteed that the sum of n over all test cases does not exceed 2⋅105.

 

Output

For each test case print a single integer  — the answer to the problem.

解题思路

另一种写法

主要思路: 把每一个数字因子中包涵的完全平方数去除，如果一段内有任意两个数字剩下的因子乘积相同，那么这两个数乘积将会是一个完全平方数，所以段数++。

代码

 

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1e7 + 10;
int sq[MAXN];
void pre()            //预处理范围内每一个数对应的除去完全平方后剩余的因子成绩
{
    for (int i = 1; i < MAXN; i++)
        sq[i] = i;                    //一开始为该数字本社
    for (int i = 2; i * i < MAXN; i++)
    {
        for (int j = i * i; j < MAXN; j += i * i)
            while (sq[j] % (i * i) == 0)    //去除完全平方因子
                sq[j] /= i * i;
    }
}
int a[MAXN];
void solve()
{
    int n, k;
    cin >> n >> k;
    set<int> st;
    int ans = 0;
    for (int i = 0; i < n; i++)
        cin >> a[i];
    for (int i = 0; i < n; i++)
    {
        a[i] = sq[a[i]];            //转换
        if (st.count(a[i]))            //如果有相同的出现过，那么增加新的一段
        {
            ans++;
            st.clear();
        }
        st.insert(a[i]);
    }
    cout << ans + 1<<"\n";            //最后一段没有统计，所以结果+1
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    pre();
    int t;
    cin >> t;
    while (t--)
    {
        solve();
    }
    return 0;
}