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

最新推荐文章于 2022-11-22 23:15:14 发布

为什么他们cf写的这么快

最新推荐文章于 2022-11-22 23:15:14 发布

阅读量106

点赞数

分类专栏：线性DP

本文链接：https://blog.csdn.net/qq_42479630/article/details/105229320

版权

线性DP 专栏收录该内容

63 篇文章 0 订阅

订阅专栏

题目链接
在这里插入图片描述

思路：考虑一下每个数对答案的贡献，对第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);
}