一个数组包含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)
本题由
@javaman 翻译。
Input
第1行:一个数N, 表示数组的长度(1 <= N <= 50000)。 第2 - N + 1行:每行1个数,对应数组中的元素Ai。(1 <= Ai <= 10^9)
Output
输出符合条件的集合数量。
Input示例
6 3 5 7 3 3 5
Output示例
14
解:直接开两个set一个从前遍历一个从后遍历,如果后面的元素在前面的元素出现过就加入后面的集合,如果后面的元素存在后面的集合中则累加个数+1,如果两个
集合相等就加上累加个数。一开始图省事直接用set是否相等来做判定条件结果T到死,因为整个集合要一一比较。。。所以用set的大小来做判定条件
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<map>
#include <set>
#include <bits/stdc++.h>
using namespace std;
const int N = 3e6+10;
typedef long long LL;
typedef pair<int,int> pi;
set<int>q1,q2;
int a[N];
struct FastIO
{
static const int S = 2*100;
int wpos;
char wbuf[S];
FastIO() : wpos(0) {}
inline int xchar()
{
static char buf[S];
static int len = 0, pos = 0;
if (pos==len)
pos = 0, len = fread(buf, 1, S, stdin);
if (pos==len) exit(0);
return buf[pos ++];
}
inline int xint()
{
int s = 1, c = xchar(), x = 0;
while(c<=32) c = xchar();
if(c=='-') s = -1, c = xchar();
for(;'0'<=c && c<='9';c=xchar()) x = x*10+c-'0';
return x * s;
}
~FastIO()
{
if(wpos) fwrite(wbuf, 1, wpos, stdout), wpos = 0;
}
}io;
int num[N];
int main()
{
int n;
scanf("%d", &n);
for(int i=1;i<=n;i++) a[i]=io.xint();
int cnt=0, ans=0;
q1.clear(),q2.clear();
int j=n;
for(int i=1;i<=n;i++)
{
q1.insert(a[i]);
while(q1.find(a[j])!=q1.end()&&j>=1)
{
num[j]=1;q2.insert(a[j]);
j--;
while(q2.find(a[j])!=q2.end()&&j>=1)
{
num[j]=num[j+1]+1;
j--;
}
}
if(q1.size()==q2.size()) ans+=num[j+1];
}
printf("%d\n",ans);
return 0;
}