码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces 567C - Geometric Progression - [map维护]

时间:2019-05-02 11:32:41      阅读:152      评论:0      收藏:0      [点我收藏+]

标签:另一个   cin   define   for   题解   bsp   记录   make   href   

题目链接:https://codeforces.com/problemset/problem/567/C

 

题意:

给出长度为 $n$ 的序列 $a[1:n]$,给出公比 $k$,要求你个给出该序列中,长度为 $3$ 的等比子序列的数目。

 

题解:

首先倒着遍历,用map记录曾经出现过的每个数字的出现次数,然后再用另一个map来记录曾经出现过的所有满足 $(x,kx)$ 的二元组的数目,最后就直接维护答案即可。

 

AC代码:

#include<bits/stdc++.h>
#define IO (ios::sync_with_stdio(0),cin.tie(0),cout.tie(0))
#define mk make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
const int maxn=2e5+10;
int n;
ll k,a[maxn];
map<ll,ll> mp1;
map<P,ll> mp2;

int main()
{
    IO;

    cin>>n>>k;

    ll mx=-1e10, mn=1e10;
    for(int i=1;i<=n;i++)
        cin>>a[i], mx=max(a[i],mx), mn=min(a[i],mn);

    ll ans=0;
    for(int i=n;i>=1;i--)
    {
        if(mn/(k*k)<=a[i] && a[i]<=mx/(k*k))
            ans+=mp2[mk(a[i]*k,a[i]*k*k)];

        if(mn/k<=a[i] && a[i]<=mx/k)
            mp2[mk(a[i],a[i]*k)]+=mp1[a[i]*k];

        mp1[a[i]]++;
    }
    cout<<ans<<endl;
}

 

Codeforces 567C - Geometric Progression - [map维护]

标签:另一个   cin   define   for   题解   bsp   记录   make   href   

原文地址:https://www.cnblogs.com/dilthey/p/10801806.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!