題目鏈接：點擊查看

題目大意：中文題

題目分析：一個星期前用單旋splay寫的，一直TLE，一氣之下用set亂搞寫過去了，今天突然想起來這個題，意識到會不會是單旋的問題，改成雙旋立馬AC，而且比set還要快，果然是模板選擇的問題

對于人和寵物兩個集合而言，我們只需要維護一個splay就足夠了，因為同一時間內(nèi)只會有一個集合的剩余，用來維護剩余的這個集合內(nèi)的元素就好了，然后每次按照題意更新答案就好了

代碼：
?

#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> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;const int mod=1000000;class Splay//存儲規(guī)則：小左大右，重復(fù)節(jié)點記錄 {#define root e[0].ch[1] //該樹的根節(jié)點private:class node{public:int v,father;//節(jié)點值，父級節(jié)點 int ch[2];//左孩子=0，右孩子=1int sum;//自己+自己下級有多少節(jié)點。在根節(jié)點為1。int recy;//記錄自己被重復(fù)了幾次};node e[N];//Splay樹主體int n,points;//使用存儲數(shù),元素數(shù)void update(int x){e[x].sum=e[e[x].ch[0]].sum+e[e[x].ch[1]].sum+e[x].recy;}int identify(int x){return e[e[x].father].ch[0]==x?0:1;}void connect(int x,int f,int son)//連接函數(shù)。用法：connect(son,father,1/0){e[x].father=f;e[f].ch[son]=x;}//作用：使得x的father=f，f的son=x.void rotate(int x){int y=e[x].father;int mroot=e[y].father;int mrootson=identify(y);int yson=identify(x);int B=e[x].ch[yson^1];connect(B,y,yson);connect(y,x,(yson^1));connect(x,mroot,mrootson);update(y);update(x);}void splay(int at,int to)//將at位置的節(jié)點移動到to位置{to=e[to].father;while(e[at].father!=to){int up=e[at].father;if(e[up].father==to) rotate(at);else if(identify(up)==identify(at)){rotate(up);rotate(at);}else{rotate(at);rotate(at);}}}int crepoint(int v,int father){n++;e[n].v=v;e[n].father=father;e[n].sum=e[n].recy=1;return n;}void destroy(int x)//pop后摧毀節(jié)點 {e[x].v=e[x].ch[0]=e[x].ch[1]=e[x].sum=e[x].father=e[x].recy=0;if(x==n) n--;//最大限度優(yōu)化}public:int getroot(){return root;}int find(int v)//用于外部的find調(diào)用{int now=root;while(true){if(e[now].v==v){splay(now,root);return now;}int next=v<e[now].v?0:1;if(!e[now].ch[next]) return 0;now=e[now].ch[next];}}int build(int v)//內(nèi)部調(diào)用的插入函數(shù)，沒有splay {points++;if(points==1)//特判無點狀態(tài) {root=n+1;crepoint(v,0);}else{int now=root;while(true)//向下找到一個空節(jié)點 {e[now].sum++;//自己的下級肯定增加了一個節(jié)點 if(v==e[now].v){e[now].recy++;return now;}int next=v<e[now].v?0:1;if(!e[now].ch[next]){crepoint(v,now);e[now].ch[next]=n;return n;}now=e[now].ch[next];}}return 0;}void push(int v)//插入元素時，先添加節(jié)點，再進(jìn)行伸展 {int add=build(v);splay(add,root);}void pop(int v)//刪除節(jié)點 {int deal=find(v);if(!deal) return;points--;if(e[deal].recy>1){e[deal].recy--;e[deal].sum--;return;}if(!e[deal].ch[0]){root=e[deal].ch[1];e[root].father=0;}else{int lef=e[deal].ch[0];while(e[lef].ch[1]) lef=e[lef].ch[1];splay(lef,e[deal].ch[0]);int rig=e[deal].ch[1];connect(rig,lef,1);connect(lef,0,1);update(lef);}destroy(deal);}int rank(int v)//獲取值為v的元素在這棵樹里是第幾小 {int ans=0,now=root;while(true){if(e[now].v==v) {ans+=e[e[now].ch[0]].sum;splay(now,root);return ans+1;}if(now==0) return 0;if(v<e[now].v) now=e[now].ch[0];else{ans=ans+e[e[now].ch[0]].sum+e[now].recy;now=e[now].ch[1];}}return 0;}int atrank(int x)//獲取第x小的元素的值 {if(x>points) return -inf;int now=root;while(true){int minused=e[now].sum-e[e[now].ch[1]].sum;if(x>e[e[now].ch[0]].sum&&x<=minused) break;if(x<minused) now=e[now].ch[0];else{x=x-minused;now=e[now].ch[1];}}splay(now,root);return e[now].v;}int upper(int v)//尋找該值對應(yīng)的一個最近的上界值 {int now=root;int result=inf;while(now){if(e[now].v>=v&&e[now].v<result) result=e[now].v;if(v<e[now].v) now=e[now].ch[0];else now=e[now].ch[1];}return result;}int lower(int v)//尋找該值對應(yīng)的一個最近的下界值 {int now=root;int result=-inf;while(now){if(e[now].v<=v&&e[now].v>result) result=e[now].v;if(v>e[now].v) now=e[now].ch[1];else now=e[now].ch[0];}return result;}#undef root }tree;int main() { #ifndef ONLINE_JUDGE // freopen("input.txt","r",stdin); // freopen("output.txt","w",stdout); #endif // ios::sync_with_stdio(false);int n;scanf("%d",&n);int cur=-1;LL ans=0;int cnt=0;for(int i=1;i<=n;i++){int op,x;scanf("%d%d",&op,&x);if(cur==-1)//沒有人或?qū)櫸飡tree.push(x);cnt++;cur=op;} else if(op==cur)//已經(jīng)有人或?qū)櫸锪?{tree.push(x);cnt++;}else//可以領(lǐng)養(yǎng) {if(tree.find(x))//存在一個相同的數(shù)tree.pop(x);else{LL ans1=tree.lower(x);LL ans2=tree.upper(x);if(llabs(ans1-x)<=llabs(ans2-x)){ans=(ans+llabs(ans1-x))%mod;tree.pop(ans1);}else{ans=(ans+llabs(ans2-x))%mod;tree.pop(ans2);}}cnt--;if(!cnt)cur=-1;}}printf("%lld\n",ans);return 0; }

?

總結(jié)

以上是生活随笔為你收集整理的HYSBZ - 1208 宠物收养所(Splay)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。