51nod 1280 前缀后缀集合（离散化、哈希、素数表）

最新推荐文章于 2022-02-26 16:52:46 发布

EnjoyingAC

最新推荐文章于 2022-02-26 16:52:46 发布

阅读量196

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/qq_37685156/article/details/82586882

版权

数论专栏收录该内容

23 篇文章 1 订阅

订阅专栏

题目链接

https://www.51nod.com/onlineJudge/submitDetail.html#!judgeId=627203

题意

一个数组包含N个正整数，其中有些是重复的。一个前缀后缀集是满足这样条件的下标对(P,S)， 0<= P,S < N 满足数组元素A[0..P]的值也在A[S..N - 1]的值中出现，并且A[S..N - 1]中的值也再A[0..P]中出现。换句话说前缀的集合A[0..P]与后缀集合A[S..N - 1]包含完全相同的值。求这样的前缀后缀集合的数量。

例如：3 5 7 3 3 5，共有14个集合符合条件：(1, 4), (1, 3), (2, 2), (2, 1), (2, 0), (3, 2), (3, 1), (3, 0), (4, 2), (4, 1), (4, 0), (5, 2), (5, 1), (5, 0)

题解

每一个前缀映射一个值，每个后缀映射一个值。根据这个映射值来判断该前缀和后缀是否包含完全相同的值。

根据每个值贡献且只贡献一次，我们可以将这N个值离散化为若干个质数。
那么，由质数的唯一分解定理可知，这样累乘质数得到的值是一一对应的。

又由于N最多可能有50000个质数，全部乘起来long long存不下。用unsigned long long去自然溢出，相当于模2^64。

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
const int maxn=5e4+7;

int prime[maxn]={2,3,5,7,11,13,17};//7个

void getPrime()
{
    int i=7,n=19;
    while(i<maxn)
    {
        int tmp=(int)sqrt(n+0.5),j;
        for(j=0;prime[j]<=tmp;j++)
        {
            if(n%prime[j]==0) break;
        }
        if(prime[j]>tmp)
        {
            prime[i++]=n;
        }
        n+=2;
    }
}

vector<int> V;
int getid(int x)
{
    return lower_bound(V.begin(),V.end(),x)-V.begin()+1;
}
int a[maxn];
map<ull,int> mp;
bool vis[maxn];
int main()
{
    int n;
    getPrime();
    while(~scanf("%d",&n))
    {
        V.clear();
        for(int i=0;i<n;i++)
        {
            int x;
            scanf("%d",&x);
            a[i]=x;
            V.push_back(x);
        }
        sort(V.begin(),V.end());
        V.erase(unique(V.begin(),V.end()),V.end());
        mp.clear();
        memset(vis,false,sizeof(vis));

        ull now=1;
        for(int i=0;i<n;i++)
        {
            int id=getid(a[i]);


            if(!vis[id]) now=now*(ull)prime[id];
            vis[id]=true;
            //cout<<"now1="<<now<<endl;
            mp[now]++;
        }

        now=1;
        long long ans=0;
        memset(vis,false,sizeof(vis));
        for(int i=n-1;i>=0;i--)
        {
            int id=getid(a[i]);
            if(!vis[id])   now=now*(ull)prime[id];
            vis[id]=true;
            //cout<<"now2="<<now<<endl;
            if(!mp[now]) continue;
            ans+=mp[now];
        }
        printf("%lld\n",ans);
    }
    return 0;
}