http://codeforces.com/contest/567/problem/C
题意:
含有n个数的数组,然后从里面找3个数ai,aj,ak,使其成为K的等比数列。
还要满足这三个数的下标是逐一上升的。
解题思路:
将中间的数组记录 x/k 和 x*k的个数。用map来维持。
写这题的题解是为了铭记一个奇淫巧计。
如果 mapmm;
mm[x*k]++的时候,当x*k很大的时候会溢出。
这时候就需要这样来化解: long long x = x*k. mm[x]++;
亲测可行。
对了上次还碰到个奇淫巧计: tr[i].val = 1LL<<(a[has[l]]);此时的tr[i].val很大,将1变成long long型就AC了。奇淫巧计啊啊啊啊啊。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2*1e5 + 100;
LL arr[N];
map<LL,LL>mm;
pair<LL,LL>pll[N];
int main()
{
int n,k; cin>>n>>k;
for(int i=1;i<=n;i++)
{
LL x; cin>>x; arr[i] = x;
LL u = 0 ; if( x%k == 0) u = mm[x/k];
LL now = x*k;
LL v = mm[now];
pll[i] = make_pair(u,v);
mm[x]++;
}
LL ans = 0;
for(int i=2;i<n;i++)
{
LL x = arr[i];
LL now = x*k;
ans += pll[i].first * (mm[now] - pll[i].second);
if( k==1 || x==0) ans -= pll[i].first;
}
long long ans1 = ans;
cout<<ans1<<endl;
return 0;
}