ACWing三元组（前后缀分解）

最新推荐文章于 2024-05-05 11:35:29 发布

追风小小少年

最新推荐文章于 2024-05-05 11:35:29 发布

阅读量166

点赞数

分类专栏：算法 c++ 文章标签： c++ 算法图论

本文链接：https://blog.csdn.net/m0_63388363/article/details/130904007

版权

算法同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

c++

9 篇文章 0 订阅

订阅专栏

ACWing三元组

题目
算法
代码

题目

在这里插入图片描述

算法

前后缀分解

代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N = 2e5+10;
#define int long long
int n,m;
int a[N];
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%lld", &a[i]);
    unordered_map<int,int> l,r;
    for(int i=1;i<=n;i++)
    {
        r[a[i]]++;    
    }
    long long res=0;
    for(int i=1;i<=n;i++)
    {
        int x=a[i];
        r[a[i]]--;
        if((x%m+m)%m==0)
        {
            res+=l[x/m]*r[x*m];
        }
        l[a[i]]++;
    }
    cout<<res;
}