[codevs 1343] 蚱蜢(省队选拔赛湖南)
http://codevs.cn/problem/1343/
題解:
本題splay基本操作:
1.如果是左跳,比如從 x 左跳到 y,就相當于查詢 [ y, x ) 區間的最大值,那么就把 y-1 伸展到根,把 x 伸展到根的右節點,那么根的右節點的左節點對應的就是這段區間。
1.如果是右跳,比如從 x 右跳到 y,就相當于查詢 ( x, y ] 區間的最大值,那么就把 x 伸展到根,把 y+1 伸展到根的右節點,那么根的右節點的左節點對應的就是這段區間。
3.這里 splay 是要將某個位置的節點伸展到給定位置,所以采用從頂向下查詢第 rank 個的方式伸展,見代碼。
4.remove操作:刪除第 k 個,就把第 k-1 個伸展到根,把第 k+1 個伸展到根的右節點,刪除它的左節點即可。別忘記 update 操作。
5.insert操作:如果要插到第 k 處,就把第 k-1 個節點旋轉到根,把第 k 個節點旋轉到根的右節點,那么插入到根的右節點的左節點就可以代替原來的第 k 個。好像沒必要像我分兩種情況。最后別忘 update,先子節點再根節點。
6.move操作:就是先 remove 再 insert。
7.宏定義:因為用到大量的根節點的左右節點的調用,而這里還不適合用引用,所以干脆定義了宏,簡單清楚,編程效果好了很多。
8.邊界防溢出:因為1、2中的將 y-1、y+1 伸展到根都是危險操作,因為 y 有可能等于 1 或 n,而 0 節點不能成為根節點,因為0節點默認為空;n+1 節點直接溢出,所以建立兩個虛擬節點,一個為1,一個為 n+1,分別放在開頭和末尾,對編程本身沒影響,但不用擔心溢出了,不過各個節點相應的位置就變了,讀入位置 x 后 x++ 就行了。
代碼:
總時間耗費: 6439ms?總內存耗費: 3 kB
有些慢,但內存耗的很少。
#include<cstdio> #include<algorithm> using namespace std;const int INF = 1e9 + 7; const int maxn = 100000 + 10;int root, ch[maxn][2], v[maxn], s[maxn], maxv[maxn];void update(int o) {maxv[o] = max(v[o], maxv[ch[o][0]]);maxv[o] = max(maxv[o], maxv[ch[o][1]]);s[o] = s[ch[o][0]] + s[ch[o][1]] + 1; }int cmp(int o, int k) {int sum = s[ch[o][0]] + 1;if(sum == k) return -1;return k < sum ? 0 : 1; }void rotate(int& o, int d) {int k = ch[o][d^1]; ch[o][d^1] = ch[k][d]; ch[k][d] = o;update(o); update(k); o = k; }void splay(int& o, int k) {int d = cmp(o, k);if(d == -1) return;if(d == 1) k -= s[ch[o][0]] + 1;int p = ch[o][d];int d2 = cmp(p, k);int k2 = (d2 == 0) ? k : k - s[ch[p][0]] - 1;if(d2 != -1) {splay(ch[p][d2], k2);if(d2 == d) rotate(o, d^1); else rotate(ch[o][d], d);}rotate(o, d^1); }void build(int l, int r, int pa) {if(l > r) return;if(l == r) { maxv[l] = v[l]; s[l] = 1; if(l < pa) ch[pa][0] = l; else ch[pa][1] = l;return;} int m = (l+r) >> 1; maxv[m] = v[m];build(l, m-1, m); build(m+1, r, m); update(m);if(m < pa) ch[pa][0] = m; else ch[pa][1] = m; }#define lc ch[root][0] #define rc ch[root][1]void move(int x, int y) {splay(root, x-1); splay(rc, x-s[lc]); //x+1 - (s[lc]+1)int o = ch[rc][0];ch[rc][0] = 0; //removeupdate(rc);if(x < y) {splay(root, y-1); splay(rc, y-s[lc]-1); //y - (s[lc]+1)ch[rc][0] = o; update(rc); update(root); //notice} else {splay(root, y); splay(lc, y-1); //ch[lc][1] = o; update(lc); update(root);} }int main() {int n, m; scanf("%d%d", &n, &m); for(int i = 2; i <= n+1; i++) scanf("%d", &v[i]); build(1, n+2, 0); root = (n+3) >> 1;for(int i = 1; i <= m; i++) {int x, k; char cmd[1]; scanf("%d%s%d", &x, &cmd, &k); x++;switch(cmd[0]) {case 'L': {splay(root, x-k-1); splay(rc, x-s[lc]-1); //x - (s[lc]+1)printf("%d\n", maxv[ch[rc][0]]); move(x, x-k);break; }case 'D': {splay(root, x); splay(rc, x+k-s[lc]); //x+k+1 - (s[lc]+1)printf("%d\n", maxv[ch[rc][0]]); move(x, x+k);break;}}}return 0; }總結
以上是生活随笔為你收集整理的[codevs 1343] 蚱蜢(省队选拔赛湖南)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: [codevs 1514] 书架
- 下一篇: [总结] 网络流最大流算法反向边的作用