题意:
首先对于一个序列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; }