训练--小菲菲爱数数

 题目链接

 

 想法:

将每个质因子它出现的下标用数组存起来。

这里以质因子2举例

0   1    2     3    4    5    6    7    8    9   10           (n个数)

                  2                      2                              假设2在这些地方出现了（下标为3，7的地方）

那么哪些子数组会包含质因子2呢?     

我是这样考虑的，【0~2】（3*（3+1）个子数组） 【4~6】（3*（3+1）个子数组） 【8~9】（同理）这三个区间的所有子数组一定都不包含2，统计出这些个数，然后用n*（n+1）/2减去这些数， 就能得到包含质因子2的子数组的个数。

可能想复杂了，先记录着吧，我要去学Java了，有更好的解法我再修改。

#include<bits/stdc++.h>
using namespace std;
const int N=200010,M=1000010,mod=1e9+7;
typedef long long LL;
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
#define endl '\n'
map<int,vector<int>>num;
int n;
int a[N];
void solve(int x,int index)
{
	for(int i=2;i<=x/i;i++)
		if(x%i==0)
		{
			while(x%i==0) x/=i;
			num[i].push_back(index);
		}
	if(x>1) num[x].push_back(index);
}
int main()
{
	ios;
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n;
		num.clear();
		for(int i=0;i<n;i++)
			cin>>a[i];
		for(int i=0;i<n;i++)
		{
			int x=a[i];
			solve(x,i);
		}
		LL res=0;
		for(auto t:num)
		{
			vector<int>&v=t.second;
			res+=(LL)n*(n+1)/2;
			for(int i=0;i<v.size();i++)
			{
				int x=v[i];
				if(i==0) res-=(LL)x*(x+1)/2;
				else
				{
					x=v[i]-v[i-1]-1;
					res-=(LL)x*(x+1)/2;
				}
			}
			int x=n-v[v.size()-1]-1;
			res-=(LL)x*(x+1)/2;
		}
		cout<<res<<endl;
	}
	return 0;
}