# P7530 [USACO21OPEN] United Cows of Farmer John P
## 题目描述
农夫约翰合牛国(The United Cows of Farmer John,UCFJ)将要选派一个代表队参加国际牛学奥林匹克(International bOvine olympIad,IOI)。
有 $N$ 头奶牛参加了代表队选拔。她们站成一行,奶牛 $i$ 的品种为 $b_i$。
代表队将会由包含至少三头奶牛的连续区间组成——也就是说,对于满足 $1\le l<r\le N$ 且 $r−l\ge 2$ 的奶牛
$l\dots r$。选定区间内的三头奶牛将会被指定为领队。出于法律原因,最边上的两头奶牛必须是领队。此外,为了避免种内冲突,每一名领队都必须与代表队的其他成员(包括领队)品种不同。
请帮助 UCFJ 求出(由于纳税原因)他们可以选派参加 IOI 的代表队的方法数。如果两个代表队拥有不同的成员或不同的领队,则被认为是不同的。
## 输入格式
输入的第一行包含 $N$。
第二行包含 $N$ 个整数 $b_1,b_2,\dots,b_N$,均在范围 $[1,N]$ 之间。
## 输出格式
输出一行一个整数表示可能的代表队的数量。
## 输入输出样例 #1
### 输入 #1
```
7
1 2 3 4 3 2 5
```
### 输出 #1
```
9
```
## 说明/提示
#### 样例解释
每一代表队对应以下的一组领队:
$$(1,2,3),(1,2,4),(1,3,4),(1,4,7),(2,3,4),(4,5,6),(4,5,7),(4,6,7),(5,6,7).$$
#### 数据范围与约定
$1\le N\le 2\times 10^5$
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const ll N=2e5;
ll n;
ll a[N+5];
ll pre[N+5],pos[N+5];
ll tr[4*N+5],lz[4*N+5];
ll ans;
void update(ll p){
tr[p]=tr[2*p]+tr[2*p+1];
return;
}
void pushdown(ll p,ll l,ll r){
if(!lz[p]) return;
ll mid=(l+r)>>1;
tr[2*p]+=(mid-l+1)*lz[p];
tr[2*p+1]+=(r-mid)*lz[p];
lz[2*p]+=lz[p];
lz[2*p+1]+=lz[p];
lz[p]=0;
return;
}
void change(ll p,ll l,ll r,ll x,ll y,ll z){
if(x>y) return;
if(x<=l && r<=y){
tr[p]+=(r-l+1)*z;
lz[p]+=z;
return;
}
pushdown(p,l,r);
ll mid=(l+r)>>1;
if(y<=mid) change(2*p,l,mid,x,y,z);
else if(x>=mid+1) change(2*p+1,mid+1,r,x,y,z);
else{
change(2*p,l,mid,x,mid,z);
change(2*p+1,mid+1,r,mid+1,y,z);
}
update(p);
return;
}
ll searchh(ll p,ll l,ll r,ll x,ll y){
if(x>y) return 0;
if(l==x && r==y) return tr[p];
pushdown(p,l,r);
ll mid=(l+r)>>1;
if(y<=mid) return searchh(2*p,l,mid,x,y);
else if(x>=mid+1) return searchh(2*p+1,mid+1,r,x,y);
else return searchh(2*p,l,mid,x,mid)+searchh(2*p+1,mid+1,r,mid+1,y);
}
int main(){
scanf("%lld",&n);
for(ll i=1;i<=n;i++){
scanf("%lld",&a[i]);
pre[i]=pos[a[i]];
pos[a[i]]=i;
}
for(ll i=1;i<=n;i++){
ans+=searchh(1,1,n,pre[i]+1,i-1);
if(pre[i]){
change(1,1,n,pre[i],pre[i],-searchh(1,1,n,pre[i],pre[i]));
change(1,1,n,pre[i]+1,i-1,1);
}
else change(1,1,n,1,i-1,1);
if(pre[pre[i]]) change(1,1,n,pre[pre[i]]+1,pre[i]-1,-1);
else if(pre[i]) change(1,1,n,1,pre[i]-1,-1);
// cout<<searchh(1,1,n,pre[i]+1,i-1)<<endl;
// for(ll j=1;j<=n;j++){
// cout<<searchh(1,1,n,j,j)<<" ";
// }
// cout<<endl;
}
printf("%lld",ans);
return 0;
}
/*
7
1 2 3 4 3 2 5
*/