Description Shaolin temple is very famous for its Kongfu monks.A lot of young men go to Shaolin temple every year, trying to be a monk there. The master of Shaolin evaluates a young man mainly by his talent on understanding the Buddism scripture, but fighting skill is also taken into account. Input There are several test cases. Output A fight can be described as two ids of the monks who make that fight. For each test case, output all fights by the ascending order of happening time. Each fight in a line. For each fight, print the new monk‘s id first ,then the old monk‘s id. Sample Input 3 2 1 3 3 4 2 0 Sample Output 2 1 3 2 4 2 |
平衡树入门题,可以用STL做。
先上STL代码,注意迭代器的使用方法。STL好多类都支持lower_bound()和upper_bound()函数:
升序排列:
iterator lower_bound( const key_type &key ): 返回一个迭代器,指向键值>= key的第一个元素。
iterator upper_bound( const key_type &key ):返回一个迭代器,指向键值> key的第一个元素。
降序排列:
iterator lower_bound( const key_type &key ): 返回一个迭代器,指向键值<= key的第一个元素。
iterator upper_bound( const key_type &key ):返回一个迭代器,指向键值< key的第一个元素。
#include <vector> #include<cstdio> #include<cstring> #include<algorithm> #include<map> using namespace std; int main() { int n,x,y; while(scanf("%d",&n)!=EOF&&n) { map<int,int> sc_id; sc_id[1000000000] = 1; sc_id[-1000000000]=1; while(n--) { scanf("%d%d",&x,&y); map<int,int>::iterator itr,itl,tmp = sc_id.lower_bound(y); itr = tmp--; itl = tmp; if(itr->first-y < y-itl->first) itl=itr; printf("%d %d\n",x,itl->second); sc_id[y]=x; sc_id[-y]=x; } } return 0; }
再看用模板过了的代码(splay):
其中id是结点里的附加信息,把它去掉就是模板了。
#include <vector> #include<cstdio> #include<cstring> #include<algorithm> #include<map> using namespace std; #define MAXN 100010 struct Node { int key, sz, cnt ,id; Node *ch[2], *pnt;//左右儿子和父亲 Node(){} Node(int x, int y, int z){ key = x, sz = y, cnt = z; } void rs() { sz = ch[0]->sz + ch[1]->sz + cnt; } }nil(0, 0, 0), *NIL = &nil; struct Splay{//伸展树结构体类型 Node *root; int ncnt;//计算key值不同的结点数,注意已经去重了 Node nod[MAXN]; void init(){// 首先要初始化 root = NIL; ncnt = 0; } void rotate(Node *x, bool d){//旋转操作,d为true表示右旋 Node *y = x->pnt; y->ch[!d] = x->ch[d]; if (x->ch[d] != NIL) x->ch[d]->pnt = y; x->pnt = y->pnt; if (y->pnt != NIL){ if (y == y->pnt->ch[d]) y->pnt->ch[d] = x; else y->pnt->ch[!d] = x; } x->ch[d] = y; y->pnt = x; y->rs(); x->rs(); } void splay(Node *x, Node *target){//将x伸展到target的儿子位置处 Node *y; while (x->pnt != target){ y = x->pnt; if (x == y->ch[0]){ if (y->pnt != target && y == y->pnt->ch[0]) rotate(y, true); rotate(x, true); } else{ if (y->pnt != target && y == y->pnt->ch[1]) rotate(y, false); rotate(x, false); } } if (target == NIL) root = x; } /************************以上一般不用修改************************/ void insert(int key,int id) {//插入一个值 if (root == NIL){ ncnt = 0; root = &nod[++ncnt]; root->ch[0] = root->ch[1] = root->pnt = NIL; root->key = key; root->id = id; root->sz = root->cnt = 1; return; } Node *x = root, *y; while (1){ x->sz++; if (key == x->key){ x->cnt++; x->rs(); y = x; break; } else if (key < x->key){ if (x->ch[0] != NIL) x = x->ch[0]; else{ x->ch[0] = &nod[++ncnt]; y = x->ch[0]; y->key = key; y->id = id; y->sz = y->cnt = 1; y->ch[0] = y->ch[1] = NIL; y->pnt = x; break; } } else{ if (x->ch[1] != NIL) x = x->ch[1]; else{ x->ch[1] = &nod[++ncnt]; y = x->ch[1]; y->key = key; y->id = id; y->sz = y->cnt = 1; y->ch[0] = y->ch[1] = NIL; y->pnt = x; break; } } } splay(y, NIL); } Node* search(int key){ //查找一个值,返回指针 if (root == NIL) return NIL; Node *x = root, *y = NIL; while (1){ if (key == x->key){ y = x; break; } else if (key > x->key){ if (x->ch[1] != NIL) x = x->ch[1]; else break; } else{ if (x->ch[0] != NIL) x = x->ch[0]; else break; } } splay(x, NIL); return y; } Node* searchmin(Node *x){//查找最小值,返回指针 Node *y = x->pnt; while (x->ch[0] != NIL){//遍历到最左的儿子就是最小值 x = x->ch[0]; } splay(x, y); return x; } void del(int key){//删除一个值 if (root == NIL) return; Node *x = search(key), *y; if (x == NIL) return; if (x->cnt > 1){ x->cnt--; x->rs(); return; } else if (x->ch[0] == NIL && x->ch[1] == NIL){ init(); return; } else if (x->ch[0] == NIL){ root = x->ch[1]; x->ch[1]->pnt = NIL; return; } else if (x->ch[1] == NIL){ root = x->ch[0]; x->ch[0]->pnt = NIL; return; } y = searchmin(x->ch[1]); y->pnt = NIL; y->ch[0] = x->ch[0]; x->ch[0]->pnt = y; y->rs(); root = y; } int rank(int key) { //求结点高度 Node *x = search(key); if (x == NIL) return 0; return x->ch[0]->sz + 1/* or x->cnt*/; } Node* findk(int kth){//查找第k小的值 if (root == NIL || kth > root->sz) return NIL; Node *x = root; while (1){ if (x->ch[0]->sz +1 <= kth && kth <= x->ch[0]->sz + x->cnt) break; else if (kth <= x->ch[0]->sz) x = x->ch[0]; else{ kth -= x->ch[0]->sz + x->cnt; x = x->ch[1]; } } splay(x, NIL); return x; } }sp; int main() { int n,x,y; while(scanf("%d",&n)!=EOF&&n) { sp.init(); sp.insert(1000000000,1); while(n--) { scanf("%d%d",&x,&y); sp.insert(y,x); int th = sp.rank(y); Node* r = sp.findk(th+1); Node* l = th>1 ? sp.findk(th-1) : r; if( y-l->key > r->key-y ) l = r; printf("%d %d\n",x,l->id); } } return 0; }