cf 567c 思维

题意：要求计算出一个长为n的序列中，长度为3的子序列且这个子序列能构成公比为k的等比数列的总个数。
假设某一个等比以p结尾，那么他的贡献，就来自于，p/k，p/k/k，那么如何同届呢。我们可以发现，以p结尾的子序列，贡献和为，在第一个p/k前面有多少个p/k/k，第二个p/k前面有多少个p/k/k，。。。。。以此类推。

    #include<iostream>
    #include<cstring>
    #include<algorithm>
    #include<cmath>
    #include<cstdlib>
    #include<climits>
    #include<stack>
    #include<vector>
    #include<queue>
    #include<set>
    #include<map>
    //#include<regex>
    #include<cstdio>
    #define up(i,a,b)  for(int i=a;i<b;i++)
    #define dw(i,a,b)  for(int i=a;i>b;i--)
    #define upd(i,a,b) for(int i=a;i<=b;i++)
    #define dwd(i,a,b) for(int i=a;i>=b;i--)
    //#define local
    typedef long long ll;
    const double esp = 1e-6;
    const double pi = acos(-1.0);
    const int INF = 0x3f3f3f3f;
    const int inf = 1e9;
    using namespace std;
    int read()
    {
        char ch = getchar(); int x = 0, f = 1;
        while (ch<'0' || ch>'9') { if (ch == '-')f = -1; ch = getchar(); }
        while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0'; ch = getchar(); }
        return x * f;
    }
    typedef pair<int, int> pir;
    #define lson l,mid,root<<1
    #define rson mid+1,r,root<<1|1
    #define lrt root<<1
    #define rrt root<<1|1
    int n, k;
    const int N = 2e5 + 10;
    ll a[N];
    map<ll, ll>mp,cnt;
    int main() {
        n = read(), k = read();
        up(i, 0, n)a[i] = read();
        if (k == 1)
        {
            ll ans = 0;
            up(i, 0, n)mp[a[i]]++;
            up(i, 0, n)if (mp[a[i]] > 2)ans += mp[a[i]] * 1ll * (mp[a[i]] - 1)*(mp[a[i]] - 2) / 6,mp[a[i]]=0;
            cout << ans << endl;
            return 0;
        }
        ll ans = 0;
        int zero = 0;
        up(i, 0, n)
        {
            cnt[a[i]]++;
            if (a[i] % k == 0 && a[i] != 0)
            {
                mp[a[i] * k] += cnt[a[i] / k];
            }
            else if (a[i] == 0)zero++;
            ans += mp[a[i]];
        }
        cout << ans + (zero) * 1ll * (zero - 1)*(zero - 2) / 6 << endl;
    }

转载于:https://www.cnblogs.com/LORDXX/p/11586931.html