题意:
首先对于一个序列X,设一个值为ans1,ans1表示把这个序列划分成的最小组数,使得每一组中任意两个数相乘是平方数
对与本题:给定原序列A,对于A的任意一个连续子序列X,都有一个ans1,问A序列所有子序列的ans1都有多少个;
输出ans[i],表示ans1为i的子序列的个数;
思路:
首先对于每个数,如果它有 平方数因子,那可以直接除掉,因为不影响,也为了方便后面计算ans1;O(n*max(a[i]));同时对这个数hash成1~5000的一个数;(开始我用map直接标记超时了);
n才5000,可以直接暴力找每个连续子序列,同时计算其ans1;
#include <bits/stdc++.h>
using namespace std;
const int maxn = 5000 + 7;
int n, a[maxn], ans[maxn], b[maxn], id;
map<int, int> mp;
int t0 = 0, t1 = 0, t2 = 0;
int cnt[maxn];
void work(int id) {
for(int i = 2; ; ++i) {
int t = i*i;
if(a[id] >= 0 && t > a[id]) return;
if(a[id] < 0 && t > -1*a[id]) return;
if(a[id]%t == 0) {
while(a[id]%t == 0) {
a[id] /= t;
}
}
}
}
void init() {
scanf("%d", &n);
id = 1;
for(int i = 1; i <= n; ++i) {
scanf("%d", &a[i]);
work(i);
int t;
if(t = mp[a[i]]) b[i] = t;
else b[i] = mp[a[i]] = id++;
}
memset(ans, 0, sizeof ans);
}
int main() {
init();
for(int i = 1; i <= n; ++i) {
for(int k = 1; k < id; ++k) cnt[k] = 0;
int sum = 0;
for(int j = i; j <= n; ++j) {
if(a[j] && !cnt[b[j]]) {
sum++;
}
cnt[b[j]]++;
ans[max(1, sum)]++;
}
}
for(int i = 1; i <= n; ++i) {
printf("%d%c", ans[i], (i == n ? '\n' : ' '));
}
return 0;
}