Codeforces Round #Pi (Div. 2) C. Geometric Progression（dp+算贡献）

题目链接


思路：考虑一下每个数对答案的贡献，对第i个数a【i】如果是k的倍数的话，那么a【i】的贡献就是前面a【i】/k的数个数，dp【i】代表【1，i-1】中a【i】/k的个数。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+1;
ll dp[maxn],a[maxn];
map<ll,ll>p;
int main()
{
	int n,k;
	scanf("%d %d",&n,&k);
	for(int i=1;i<=n;++i)
	{
		scanf("%lld",&a[i]);
		if(a[i]%k==0) dp[i]=p[a[i]/k];
		else dp[i]=0;
		p[a[i]]++;
	}
	p.clear();
	ll ans=0;
	for(int i=1;i<=n;++i)
	{
		if(a[i]%k==0) ans+=p[a[i]/k];
		p[a[i]]+=dp[i];	
	}
	printf("%lld\n",ans);
}