非诚勿扰 - HDU 4557 - Treap入门

链接:

  http://acm.hdu.edu.cn/showproblem.php?pid=4557

题目：

Problem Description

　　作为2013年699万应届毕业生中的一员，由于宏观经济的不景气，小明在毕业当天就华丽丽地失业了！

　　经历了千难万苦的求职过程，小明特别能理解毕业生的就业之难，所以，他现在准备创建一家专门针对IT人才的求职中介公司——非诚勿扰人力资源开发有限公司。

　　基于工作的需要，小明根据求职学生的简历描述为每人评定了一个综合能力值，能力值是一个小于等于20的正整数，值越高表示能力越强。当有公司试图招聘IT人员的时候（每次只招聘1名），需要提出一个综合能力的最低需求，若人才库中有符合要求的人才，则一定能成功招聘。当然，若有多名学生同时满足招聘公司的需求，鉴于高能力人才的稀缺，小明总是优先把能力值低的人才推荐过去；如果依然有多名人员符合要求，则小明就把其中最早来求职的那位学生推荐过去。

　　需要说明的是，刚开始的时候，公司的人才库为空，而且一名学生只能和一个企业签约，如果推荐成功，则该名学生的信息需要从人才库中删除。

Input

　　输入数据的第一行是一个正整数T（1 <= T <= 20）, 表示有T组测试数据；

　　每组测试数据第一行是一个整数n（0 <= n <= 1000），表示按照时间先后发生了n次事件。接下来的n行，每行描述一次事件。对于一次事件，先是一个字符串”Add”或者”Find”，其中”Add”表示有一名学生加入了人才库，”Find”表示有企业想招聘一名人员。

如果字符串是”Add”，则后面将有一个字符串s和一个数字d，用空格隔开，分别表示该名学生的名字和综合能力值，名字由小写字母组成，不为空且长度不超过15；如果字符串是”Find”，则后面将有一个数字，表示招聘公司对人才综合能力的最低要求。

Output

对于每组测试数据，第一行输出”Case #c:”（不包含引号）

c是测试数据的组数，从1开始。

然后输出n行，表示n次事件的结果

如果本次事件是添加人才信息入库，则请输出加入该信息后，人才库内的人员数量；

如果本次事件是企业来招聘，则请输出将被录用的人才名字，如果没有人才符合要求，就请输出”WAIT…”

Sample Input

1

5

Add lcy 1

Add lyd 19

Find 11

Find 13

Add zxs 10

Sample Output

Case #1:

1

2

lyd

WAIT…

2

思路：

  改一下搜索函数，记录下第一个大于等于题目所给的权就可以，记得能力值相同的情况下来得晚的位次就靠后。

实现：

#include <iostream>
#include <algorithm>
#include <set>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <list>
#include <iomanip>
#include <functional>
#include <sstream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>

#define read read()
#define edl putchar('\n')
#define clr(a, b) memset(a,b,sizeof a)

using namespace std;
int Size;
struct node {
node *ch[2];
string name;
int v, fix, date, size;
node(string name, int v, int date):name(name), v(v),fix(rand()), date(date), size(1) {
ch[0] = ch[1] = NULL;
}
int compare(int v_, int date_) {
if(v == v_) {
if(date == date_) return -1;
return date_ < date ? 0 : 1;
}
return v_ < v ? 0 : 1;
}
void maintain() {
size = 1;
if(ch[0] != NULL) size += ch[0]->size;
if(ch[1] != NULL) size += ch[1]->size;
}
}*root;

void rotate(node *&t, int d) {
if(t == NULL) return ;
node *tmp = t->ch[d^1];
t->ch[d^1] = tmp->ch[d];
tmp->ch[d] = t;
t->maintain();
tmp->maintain();
t = tmp;
}

void insert(node *&t, string name, int v, int date) {
if(t == NULL) {
t = new node(name, v, date);
Size++;
}
else {
int d = t->compare(v, date);
insert(t->ch[d], name, v, date);
if(t->ch[d]->fix > t->fix) rotate(t, d^1);
}
t->maintain();
}

void remove(node *&t, int v, int date) {
if(t == NULL) return ;
int d = t->compare(v, date);
if(d == -1) {
Size--;
node *tmp = t;
if(t->ch[0] == NULL) {
t = t->ch[1];
delete tmp;
} else if(t->ch[1] == NULL) {
t = t->ch[
4000
0];
delete tmp;
} else {
int k = t->ch[0]->fix > t->ch[1]->fix ? 1 : 0;
rotate(t,k);
remove(t->ch[k],v,date);
}
} else remove(t->ch[d], v, date);
if(t != NULL) t->maintain();
}

void findsuc(int v, node *&ret) {
node *t = root;
ret = NULL;
while(t != NULL) {
if(t->v >= v) {
ret = t;
t = t->ch[0];
} else t = t->ch[1];
}
}

void print(node *t) {
if(t == NULL) return ;
print(t->ch[0]);
cout << t->name << ' ';
print(t->ch[1]);
}

void clear(node *&t) {
if(t == NULL) return;
if(t->ch[0] != NULL) clear(t->ch[0]);
if(t->ch[1] != NULL) clear(t->ch[1]);
delete t;
t = NULL;
}

int main() {
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
#endif
int t, v, n;
string str;
node *tmp;
cin >> t;
for(int cas = 1 ; cas <= t ; cas++) {
Size = 0;
clear(root);
cout << "Case #" << cas << ":\n";
cin >> n;
for(int i=1 ; i<=n ; i++){
cin >> str;
if(str == "Find") {
cin >> v;
findsuc(v,tmp);
if(tmp == NULL) {
puts("WAIT...");
} else {
cout << tmp->name << '\n';
remove(root, tmp->v, tmp->date);
}
} else {
cin >> str >> v;
insert(root, str, v, i);

if(root == NULL) puts("0");
else cout << root->size << '\n';
}
}
}
return 0;
}