思路:枚举所有的前缀后缀区间,在遍历后缀时要保证前缀区间一定包含后缀区间,可以用set和map两种方式求解,set较易理解,但map效率较高一点。
一,set:由于枚举时以保证前缀一定包含后缀,因此只要前缀后缀的set大小相同即为相同区间,因此可以保存后缀相同区间的个数,每次枚举前缀时加上后缀相同区间的个数即可
二,map: 同样保证前缀包含后缀,但是记录前缀比后缀多出的元素个数t,以及后缀相同区间的个数sum,当前缀有新元素时,重置sum=0(因为此时后缀肯定没有和前缀相同的区间),然后遍历后缀a[R],若map[a[R]]==1则t--,imap[a[R]]=2,当t==0时则说明此时前缀后缀区间相同,因此ans++,sum++,如此反复即可
Code 1:
#include<iostream>
#include<set>
using namespace std;
typedef long long LL;
const int MAX_N=50005;
int n;
int a[MAX_N];
int sum[MAX_N];
set<int> L,R;
int main()
{
ios::sync_with_stdio(false);
cin>>n;
for(int i=0;i<n;++i)
cin>>a[i];
int l=0,r=n-1;
LL ans=0;
while(l<n){
L.insert(a[l++]);
while(r>=0&&L.find(a[r])!=L.end()){
R.insert(a[r--]);
sum[R.size()]++;
}
ans+=sum[L.size()];
}
cout<<ans<<endl;
return 0;
}
Code 2:
#include<iostream>
#include<map>
using namespace std;
typedef long long LL;
const int MAX_N=50005;
int n;
int a[MAX_N];
map<int,int> imap;
int main()
{
ios::sync_with_stdio(false);
cin>>n;
for(int i=0;i<n;++i)
cin>>a[i];
//t:前缀的新元素个数 sum:后缀相同区间个数
int L=0,R=n-1,sum=0,t=0;
LL ans=0;
while(L<n){
if(imap[a[L]]){
ans+=sum;
}else{ //前缀新元素
imap[a[L]]=1;
sum=0; t++;
while(R>=0&&imap[a[R]]){
if(imap[a[R]]==1){
imap[a[R]]=2;
t--;
}
if(!t){
ans++; sum++;
}
R--;
}
}
L++;
}
cout<<ans<<endl;
return 0;
}