(CodeForces) C. Geometric Progression

倍数序列计数算法

最新推荐文章于 2020-03-21 14:17:35 发布

原创最新推荐文章于 2020-03-21 14:17:35 发布 · 385 阅读

0 ·

CC 4.0 BY-SA版权

CodeForces 专栏收录该内容

43 篇文章

订阅专栏

本文介绍了一种算法，用于计算满足特定倍数条件的长度为3的序列数量。通过使用两个map数据结构来记录数字出现的次数，该算法能够高效地解决倍数序列计数问题，特别注意了k为1和元素为0的特殊情况。

传送门

题目大意：求满足倍数是k，长度为3的序列有多少个。

解题思路：对于一个a[i],我们只要知道他的前面有多少个a[i]/k,后面有多少个a[i]*k就行了呀，所有我们用两个map维护，某个数字总共出现了多少次，和已经出现了多少次就行了，然后就是要注意一下，k为1和a[i]为0的情况，这地方我还wa了几发。

#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define fi first
#define se second
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define sc(n) scanf("%d",&n)
#define SC(n,m) scanf("%d %d",&n,&m)
#define sz(a) int((a).size())
#define rep(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const double PI=acos(-1.0);
const double eps=1e-9;
const int maxn=2e5+5;
ll n,k,a[maxn];
map<ll,ll> all,front; //所有的，前面的 
int main() {
	std::ios::sync_with_stdio(0);
	cin>>n>>k;
	rep(i,1,n)	cin>>a[i],all[a[i]]++;
	ll ans=0;
	if(k==1) {
		rep(i,1,n){
			ans+=front[a[i]]*(all[a[i]]-1-front[a[i]]);
			front[a[i]]++;
		}
	} else {
		rep(i,1,n) {
			if(a[i]%k==0) {
				if(a[i]!=0) ans+=(front[a[i]/k]*(all[a[i]*k]-front[a[i]*k]));
				else	ans+=(front[0]*(all[0]-front[0]-1));
			}
			front[a[i]]++;
		}
	}
	cout<<ans<<endl;
	return 0;
}