題目鏈接：點擊查看

題目大意：給出一個長度為 $n$ 的數列和一個整數 $k$，現在有 $q$ 次詢問，每次詢問給出一個區間 $[l,r]$，設 $cn{t}_i$ 為數字 $i$ 在區間內出現的次數，需要輸出 $\sum min(cn{t}_i,k)$，其中 $i$ 是區間內所有出現過的數字，所有詢問強制在線

題目分析：看到強制在線加上區間問題不難想到主席樹，有個思維點必須要想到才行，一開始我想的是對于每個種類的數字維護一個前綴和，然后對于每個詢問按照出現次數大于等于 $k$ 和小于 $k$ 分類討論，很可惜復雜度不是很合適

所以這個題需要想到，應該對于每個數字單獨討論，假設 $b_i$ 是 $a_i$ 后面第 $k$ 次出現 $a_i$ 的位置，那么假設需要詢問的區間為 $[l,r]$，不難看出只有當 $b_i>r$ 時，才可以選擇 $a_i$，否則會與 $b_i$ 位置的數字沖突，所以我們用主席樹維護一下 $b_i$，每次查詢只需要輸出區間 $[l,r]$ 內有多少個 $b_i>r$ 即可

代碼：

// #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; template<typename T> inline void read(T &x) {T f=1;x=0;char ch=getchar();while(0==isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(0!=isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();x*=f; } template<typename T> inline void write(T x) {if(x<0){x=~(x-1);putchar('-');}if(x>9)write(x/10);putchar(x%10+'0'); } const int inf=0x3f3f3f3f; const int N=1e6+100; int a[N],b[N]; vector<int>node[N]; struct Node {int l,r;int sum; }tree[N*20]; int cnt,root[N]; void init() {root[0]=0;tree[0].l=tree[0].r=tree[0].sum=0;cnt=1; } void update(int num,int &k,int l,int r) {tree[cnt++]=tree[k];k=cnt-1;tree[k].sum++;if(l==r)return;int mid=(l+r)>>1;if(num<=mid)update(num,tree[k].l,l,mid);elseupdate(num,tree[k].r,mid+1,r); } int query(int i,int j,int L,int R,int l,int r)//[L,R]:當前區間,[l,r]:目標區間 {if(L>r||R<l)return 0;if(L>=l&&R<=r)return tree[j].sum-tree[i].sum;int mid=(L+R)>>1;return query(tree[i].l,tree[j].l,L,mid,l,r)+query(tree[i].r,tree[j].r,mid+1,R,l,r); } int main() { #ifndef ONLINE_JUDGE // freopen("data.in.txt","r",stdin); // freopen("data.out.txt","w",stdout); #endif // ios::sync_with_stdio(false);init();int n,k;read(n),read(k);for(int i=1;i<=n;i++)read(a[i]);for(int i=n;i>=1;i--){if((int)node[a[i]].size()<k)b[i]=n+1;elseb[i]=node[a[i]][(int)node[a[i]].size()-k];node[a[i]].push_back(i);}for(int i=1;i<=n;i++){root[i]=root[i-1];update(b[i],root[i],1,n+1);}int ans=0,m;read(m);while(m--){int l,r;read(l),read(r);l=(l+ans)%n+1;r=(r+ans)%n+1;if(l>r)swap(l,r);write(ans=query(root[l-1],root[r],1,n+1,r+1,n+1));puts("");}return 0; }

總結

以上是生活随笔為你收集整理的CodeForces - 813E Army Creation(主席树+思维)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。