Geometric Progression(思维)

【问题】

给你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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

浅梦曾倾

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值