題目鏈接:點擊查看
題目大意:給出一個長度為 n 的數列 a ,現在規定 a 中的一段區間?[ l , r ],如果其中每個數字要么出現 0 次要么出現恰好 3 次,就規定這是個好區間,問共有多少個好區間
題目分析:可以枚舉其中一個端點,然后計算另一個端點的貢獻,我使用的是順序枚舉右端點來求出左端點的可行區間,先設 last[ a[ i ]?][ j ] 代表的是,到了第 i 個位置為止,前面第 j 個 a[ i ] 的位置,顯然 last[ a[ i ] ][ 0 ] = i ,這樣的話不難看出只有 last[ a[ i ] ][ 3 ] + 1 ~ last[ a[ i ] ][ 2 ] 這段區間是符合 a[ i ] 恰好出現三次的這個限制,稱之為合法區間,其余位置都稱之為非法區間,換句話說,對于第 i 個位置而言,只有在合法區間內的數才可以當右端點為 i 時作為左端點與其匹配,對于每個右端點 i 都會有一段相應的左端點區間,現在我們需要求出所有右端點的合法區間的交集,才能作為當前右端點的可行區間,舉個很簡單的例子,1 2 2 1 1 這個長度為 5 的數列,現在枚舉的右端點是 a[ 5 ] ,也就是最后的那個 1 ,與其對應的可行區間是第一個位置 a[ 1 ] ,但是如果第一個位置作為左端點與其匹配的話,會導致 a[ 2 ] 和 a[ 3 ] 的這兩個位置的 2 不滿足條件了,所以對于每個位置的左端點,必須要選擇前面所有位置可行區間的交集才行
關于每次更新每個位置的合法與非法區間,可以類似于滑動窗口一樣滑過去,簡單畫個圖表示一下,下圖中綠色的區間是需要實時維護的,藍色區間是上一次維護的,紅色區間是本次需要維護的
這樣我們就用線段樹來維護一下區間的覆蓋問題,合法區間不予以覆蓋,非法區間予以覆蓋,這樣最后答案就是所有沒被覆蓋到的位置了,關于區間覆蓋問題直接用線段樹的標記永久化維護即可
代碼:
?
//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<unordered_map>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=5e5+100;int last[N][5];//last[i][j]:第i個位置前面的第j個a[i]的位置 struct Node
{int l,r;int sum,lazy;
}tree[N<<2];void pushup(int k)
{if(tree[k].lazy)tree[k].sum=0;else if(tree[k].l==tree[k].r)tree[k].sum=1;elsetree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}void build(int k,int l,int r)
{tree[k].l=l;tree[k].r=r;tree[k].sum=tree[k].lazy=0;if(l==r)return;int mid=l+r>>1;build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}void update(int k,int l,int r,int val)
{if(tree[k].r<l||tree[k].l>r)return;if(tree[k].l>=l&&tree[k].r<=r){tree[k].lazy+=val;pushup(k);return;}update(k<<1,l,r,val);update(k<<1|1,l,r,val);pushup(k);
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("data.in.txt","r",stdin);
// freopen("data.out.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);int n;scanf("%d",&n);build(1,1,n);LL ans=0;for(int i=1;i<=n;i++){int x;scanf("%d",&x);for(int j=4;j>0;j--)last[x][j]=last[x][j-1];last[x][0]=i;update(1,i,i,0);//更新第i個位置這條鏈上的值 update(1,last[x][1]+1,last[x][0],1);if(last[x][3]+1<=last[x][2])update(1,last[x][3]+1,last[x][2],-1);if(last[x][4]+1<=last[x][3])update(1,last[x][4]+1,last[x][3],1);ans+=tree[1].sum;}printf("%lld\n",ans);return 0;
}
?
超強干貨來襲 云風專訪:近40年碼齡,通宵達旦的技術人生
總結
以上是生活随笔為你收集整理的CodeForces - 1418G Three Occurrences(线段树-标记永久化)的全部內容,希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯,歡迎將生活随笔推薦給好友。