題目鏈接:點(diǎn)擊查看
題目大意:給出一個(gè)長(zhǎng)度為 n 的數(shù)列 a ,再給出 m 次詢問,每次詢問給出一個(gè)區(qū)間 [ l , r ] ,問區(qū)間 [ l , r ] 內(nèi)首次出現(xiàn)的數(shù)字的位置的中位數(shù)
題目分析:題目可能比較繞,但是涉及到區(qū)間內(nèi)第 k 大的問題不難想到主席樹了,現(xiàn)在問題就是思考如何利用首次出現(xiàn)的數(shù)字這個(gè)條件
如果換種問法,問最后一次出現(xiàn)數(shù)字的位置,我們不難聯(lián)想到主席樹解決區(qū)間內(nèi)有多少個(gè)本質(zhì)不同的數(shù)字這道題目,這道題目實(shí)質(zhì)上就是維護(hù)每個(gè)數(shù)字最后一次出現(xiàn)的位置,所以回到這個(gè)題目來,首次出現(xiàn)的位置,我們可以將整個(gè)數(shù)列倒過來維護(hù),這樣根 root[ i ] 代表的那個(gè)權(quán)值線段樹的意義就是,區(qū)間 [ i?, n ] 內(nèi)每個(gè)數(shù)字首次出現(xiàn)的位置了,對(duì)于區(qū)間 [ l , r ] ,我們可以在 root[ l ] 這棵權(quán)值線段樹上,查找區(qū)間 [ l , r ] 內(nèi)有多少個(gè)不同的數(shù),再求第 k 大就好了
代碼: ?
#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>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=2e5+100;struct Node
{int l,r;int sum;
}tree[N*50];int cnt,root[N],ans[N],a[N],pre[N];void init(int pos)
{memset(pre,0,sizeof(pre));root[pos]=0;tree[pos].l=tree[pos].r=tree[pos].sum=0;cnt=1;
}void change(int pos,int &k,int l,int r,int val)
{tree[cnt++]=tree[k];k=cnt-1;tree[k].sum+=val;if(l==r)return;int mid=l+r>>1;if(pos<=mid)change(pos,tree[k].l,l,mid,val);elsechange(pos,tree[k].r,mid+1,r,val);
}int query_sum(int rt,int l,int r,int L,int R)
{if(R<l||L>r)return 0;if(L>=l&&R<=r)return tree[rt].sum;int mid=L+R>>1;return query_sum(tree[rt].l,l,r,L,mid)+query_sum(tree[rt].r,l,r,mid+1,R);
}int query_kth(int rt,int l,int r,int k)
{if(l==r)return l;int mid=l+r>>1;if(tree[tree[rt].l].sum>=k)return query_kth(tree[rt].l,l,mid,k);elsereturn query_kth(tree[rt].r,mid+1,r,k-tree[tree[rt].l].sum);
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("input.txt","r",stdin);
// freopen("output.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);int w;cin>>w;int kase=0;while(w--){int n,m;scanf("%d%d",&n,&m);init(n+1);for(int i=1;i<=n;i++)scanf("%d",a+i);for(int i=n;i>=1;i--){root[i]=root[i+1];if(pre[a[i]])change(pre[a[i]],root[i],1,n,-1);change(i,root[i],1,n,1);pre[a[i]]=i;}for(int i=1;i<=m;i++){int l,r;scanf("%d%d",&l,&r);l=(l+ans[i-1])%n+1;r=(r+ans[i-1])%n+1;if(l>r)swap(l,r);int k=(query_sum(root[l],l,r,1,n)+1)/2;ans[i]=query_kth(root[l],1,n,k);}printf("Case #%d:",++kase);for(int i=1;i<=m;i++)printf(" %d",ans[i]);puts("");}return 0;
}
?
總結(jié)
以上是生活随笔 為你收集整理的HDU - 5919 Sequence II(主席树+思维) 的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔 網(wǎng)站內(nèi)容還不錯(cuò),歡迎將生活随笔 推薦給好友。