【问题】
给你n个数,让你找在序列中有多少个等比数列 a i a_i ai, a j a_j aj, a k a_k ak,i < j < k,公比是q
【输入】
输入n和q
输入n个数【输出】
输出等比数列数目
【思路】
说实话刚开始想的是dfs之类的来做,但是看了眼数据肯定会超时……
观摩了一下大佬的做法,枚举 a j a_j aj,也就是等比数列中间那个数,然后就是 a i a_i ai的数目✖ a j a_j aj的数目,但是要注意顺序,所以就用到两个数组,(来自大佬的一(亿)点小技巧)
这个题数据太大用到了map,→_→想不到
【源代码】
#include<iostream>
#include<cstdio>
#include<string>
#include<cstdlib>
#include<cmath>
using namespace std;
#include<stack>
#include<cstdlib>
#include<string>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<algorithm>
#include<map>
#define ll long long
const int maxn = 200100;
map<ll, ll> m1;
map<ll, ll> m2;
ll a[maxn];
int main()
{
int n, k;
cin >> n >> k;
for (ll i = 1; i <= n; i++)
{
cin >> a[i];
m2[a[i]]++;
}
ll ans = 0;
for (int i = 1; i < n; i++)
{
ll v = a[i];
m2[a[i]]--;//亿点小技巧,自减之后下一步×的时候就不会出现问题了,消除前面数的影响
if (v % k == 0)
ans += m1[v / k] * m2[v * k];
m1[a[i]]++;//出现过才能用这个数
}
cout << ans << endl;
return 0;
}