CF 567C. Geometric Progression

题目链接

---

题意：

给出一个序列,求公比为k，且长度为3的子等比序列个数。(1 ≤ n, k ≤ 2e5)( - 1e9 ≤ ai ≤ 1e9)

---

解法:

dp[0][x]表示当前x的数量。
dp[1][x]表示当前以值x结尾，长度为2且公比为K的等比数列数量。

---

代码：

#include<cstring>
#include<iostream>
#include<map>
using namespace std;

#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mem(a,x)  memset(a,x,sizeof a)
typedef long long ll;
const int maxn=2e5+0.5    ;

int n,K;
map<int,ll >dp[2];
int a[maxn+10];

void solve()
{
    dp[0].clear();
    dp[1].clear();
    ll ans=0;
    for1(i,n)
    {
        int x=a[i];
        if(x%K==0)
        {
            ans+=dp[1][x/K];
            dp[1][x]+=dp[0][x/K];
        }
        dp[0][x]++;
    }
    cout<<ans<<endl;

}
int main()
{
   std::ios::sync_with_stdio(false);
   while(cin>>n>>K)
   {
       for1(i,n) cin>>a[i];
       solve();
   }
   return 0;
}

#include<cstring>
#include<iostream>
#include<map>
using namespace std;

#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
#define mem(a,x)  memset(a,x,sizeof a)
typedef long long ll;
const int maxn=2e5+0.5    ;

int n,K;
map<int,int >mp;
int cnt[maxn+10],a[maxn+10];
ll dp[maxn+10];
int ID(int x)
{
    if(!mp.count(x))
    {
        int k=mp.size();
        mp[x]=k;
        return k;
    }
    return mp[x];

}
void solve()
{
    mp.clear();
    mem(cnt,0);
    mem(dp,0);
    ll ans=0;
    for1(i,n)
    {
        int x=a[i],y=ID(x);
        if(x%K==0)
        {

            int t=ID(x/K);//这里不能直接用mp
            ans+=dp[t];

            dp[y]+=cnt[t];
        }
        cnt[y]++;
    }
    cout<<ans<<endl;

}
int main()
{
   std::ios::sync_with_stdio(false);
   while(cin>>n>>K)
   {
       for1(i,n) cin>>a[i];
       solve();
   }
   return 0;
}