一个数组包含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
李陶冶
(题目提供者)
题目不难,思路滤清即可
首先两个集合的元素要求完全相同,即A中的任意元素在B中都出现过,同理B也一样,但是怎样找到这样两个集合呢?
从前缀后缀这里出发,首先想到假如B中的某一元素在A中未曾出现,呢B集合就没必要再扩充了,这是其中一个剪枝,
然后要(
想到一个集合里的元素个数增加但是种类数不增加事答案应该怎样变化,这里难倒我了,当时思路混乱,导致无
从下手,看完题解后发现根本没有那么复杂,每次增加元素在不改变种类数的情况下只用让当前方案数加1即可。。。
)
拓展(小知识点):
map中元素的查找:
find()函数返回一个迭代器指向键值为key的元素,如果没找到就返回指向map尾部的迭代器。
map<int ,string >::iterator l_it;;
l_it=maplive.find(112);
if(l_it==maplive.end())
cout<<"we do not find 112"<<endl;
else cout<<"wo find 112"<<endl;
代码如下:
#include<stdio.h>
#include<algorithm>
#include<map>
using namespace std;
#define maxn 50005
typedef long long ll;
ll a[maxn],b[maxn],ans;
map<ll,int>p,q;
int main()
{
ll i,n,j,temp=0,res=0;
scanf("%lld",&n);
for(i=0;i<n;i++)
{
scanf("%lld",&a[i]);
b[n-i-1]=a[i];
}
for(i=0;i<=n;i++)
{
map<ll,int>::iterator it=p.find(a[i]);
if(it!=p.end())
{
ans+=res;;
continue;
}
p[a[i]]=1;
bool flag=0;
res=0;
for(j=temp;j<n;j++)
{
map<ll,int>::iterator it=p.find(b[j]);
if(it==p.end())
break;
if(flag)
{
res++;
continue;
}
q[b[j]]=1;
if(p.size()==q.size())
{
flag=1;
res++;
temp=j;
}
}
ans+=res;
}
printf("%lld\n",ans);
}