比赛时虽然贪了心,不过后面没想到怎么处理和set的排序方法忘了- -,其实是和优先队列的仿函数一样的。。。
比赛后用set pair过了。。。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 200005;
const ll mod = 1e9 + 7;
const double eps = 1e-12;
ll a[N];
struct node1 {
ll d, dm, pos;
}f[N];
struct node2 {
ll d, pos;
}g[N];
bool cmp1(node1 a, node1 b) {
return a.d > b.d;
}
bool cmp2(node2 a, node2 b) {
return a.d > b.d;
}
ll l[N], r[N], ans[N];
set<pair<ll, ll> >s;
int main() {
ll n, m;
cin >> n >> m;
for (ll i = 1;i <= n;i++) {
cin >> l[i] >> r[i];
if (i >= 2) {
f[i - 1].d = l[i] - r[i - 1];
f[i - 1].dm = r[i] - l[i - 1];
f[i - 1].pos = i - 1;
}
}
for (ll i = 1;i <= m;i++) {
cin >> a[i];
g[i].d = a[i];
g[i].pos = i;
}
sort(f + 1, f + n, cmp1);
sort(g + 1, g + m + 1, cmp2);
ll flag = 1, j = 1;
for (int i = 1;i <= m;i++)s.insert(make_pair(g[i].d,g[i].pos));
for (ll i = 1;i < n;i++){
auto it = s.upper_bound(make_pair(f[i].dm,1e18));
if (it != s.begin())it--;
/*while (it!=s.end()&& (!(g[it->second].d >= f[i].d&&g[it->second].d <= f[i].dm))) {
j++;
}*/
if(it!=s.end()&&it->first >= f[i].d&&it->first <= f[i].dm){
ans[f[i].pos] = it->second;
s.erase(it);
}
else {
flag = 0;
break;
}
}
if (flag == 0)puts("No");
else {
puts("Yes");
for (ll i = 1;i < n;i++)
cout << ans[i] << " ";
}
return 0;
}
然后又作死,自己脑补平衡树套平衡树用的splay过了,两个都错在过24组,set里错24组因为不知道排序内部到底会有什么鬼,splay二分查找时要保存满足条件的最大值,这里不是循环到最后的y值可能y会由一个相等的值转到一个小于或大于的值,因为没判断想等跳出,后面判断了为更大的才赋值。
其实这里本来只用写单层splay就可以了,另一个用个vector保存就可以了,不过想练练splay了。。。其实和普通平衡树差不多,只不过每一个节点又是一课平衡树,操作的时候注意第一个点建树为空删树,把操作部分变为对子树中节点操作,当然这个题没有复杂的up或者down。另外要注意要先声明一些东西,不然后面调用前面,前面又调用后面- -。。。
另外两种时间好似差不多- -。估计c++11中stl优化的比较好吧-w-。
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N = 800005;
const ll maxn = 800005;
const ll mod = 1e9 + 7;
const double eps = 1e-12;
ll a[N];
struct node {
ll d, dm, pos;
}f[N];
pair<ll, ll>g[N];
bool cmp(node a, node b) {
return a.d > b.d;
}
struct SplayTree;
struct SplayTree *cstm;
ll l[N], r[N], ans[N];
struct Node {
ll key;
SplayTree *st;
Node *c[2], *p;
}mem[N], *cur;
//Initialize functions without memory pool
//Splay tree
struct SplayTree {
Node *root,*nil;
void Rotate(Node *x, ll f) {
if (x == nil) return;
Node *y = x->p;
y->c[f ^ 1] = x->c[f], x->p = y->p;
if (x->c[f] != nil)
x->c[f]->p = y;
if (y->p != nil)
y->p->c[y->p->c[1] == y] = x;
x->c[f] = y, y->p = x;
}
void Splay(Node *x, Node *f) {
while (x->p != f) {
Node *y = x->p;
if (y->p == f)
Rotate(x, x == y->c[0]);
else {
ll fd = y->p->c[0] == y;
if (y->c[fd] == x)
Rotate(x, fd ^ 1), Rotate(x, fd);
else
Rotate(y, fd), Rotate(x, fd);
}
}
if (f == nil)
root = x;
}
Node *newNode(ll v, Node *p, Node *nil) {
cur->c[0] = cur->c[1] = nil;
cur->p = p;
cur->key = v;
cur->st = cstm++;
return cur++;
}
void Insert(ll v) {
Node *x = root, *y = nil;
while (x != nil) {
y = x;
x = x->c[v >= x->key];
}
y->c[v >= y->key] = x = newNode(v, y, nil);
Splay(x, nil);
}
void Insert(ll v,ll pos) {
Node *x = root, *y = nil;
while (x != nil&&x->key != v) {
y = x;
x = x->c[v >= x->key];
}
if (x == nil) {
y->c[v >= y->key] = x = newNode(v, y, nil);
cur->c[0] = cur->c[1] = nullptr;
cur->p = cur;
cur->key = 0;
cur->st = cstm++;
x->st->nil = cur++;
x->st->root = x->st->nil;
}
x->st->Insert(pos);
Splay(x, nil);
}
pair<ll, ll> fun(ll v) {
Node *x = root, *y = nil;
while (x!=nil) {
if (x->key <= v&&x->key > y->key)
y = x;
x = x->c[v > x->key];
}
if(y!=nil)
return make_pair(y->key, y->st->root->key);
return make_pair(-1, -1);
}
void Del() {
Node *x = root;
if (x->c[0] == nil&&x->c[1] == nil)root = nil;
else if (x->c[0] == nil) {
x = x->c[1];
while (x->c[0] != nil)
x = x->c[0];
Splay(x, nil);//x->c[0]->p = nil;//x->c[0]->c[0] = x->c[0]->c[1] = nil;
x->c[0] = nil;
}
else if (x->c[1] == nil) {
x = x->c[0];
while (x->c[1] != nil)
x = x->c[1];
Splay(x, nil);//x->c[1]->p = nil;//x->c[1]->c[0] = x->c[1]->c[1] = nil;
x->c[1] = nil;
}
else {
x = x->c[1];
while (x->c[0] != nil)
x = x->c[0];
Node *y = root;
Splay(x, nil);
Splay(y, x);
x->c[0] = x->c[0]->c[0];
x->c[0]->p = x;
}
//Splay(x->c[0], x->c[1]);
}
void Remove(ll v) {
Node *x = root;
while (x->key != v) {
x = x->c[v >= x->key];
}
Splay(x, nil);
Del();
}
void Remove(pair<ll, ll> it) {
Node *x = root;
while (x->key != it.first) {
x = x->c[it.first >= x->key];
}
Splay(x, nil);
x->st->Remove(it.second);
if (x->st->root == x->st->nil)
Del();
}
}s,stm[N];
void Init() {
cur = mem;
cstm = stm;
}
int main() {
ll n, m;
cin >> n >> m;
for (ll i = 1;i <= n;i++) {
cin >> l[i] >> r[i];
if (i >= 2) {
f[i - 1].d = l[i] - r[i - 1];
f[i - 1].dm = r[i] - l[i - 1];
f[i - 1].pos = i - 1;
}
}
for (ll i = 1;i <= m;i++) {
cin >> a[i];
g[i].first = a[i];
g[i].second = i;
}
sort(f + 1, f + n, cmp);
sort(g + 1, g + m + 1);
ll flag = 1, j = 1;
Init();cur->c[0] = cur->c[1] = nullptr;cur->key = 0;cur->p = cur;cur->st = cstm++;s.nil = cur++;s.root = s.nil;
for (ll i = m;i >= 1;i--)s.Insert(g[i].first,g[i].second);
for (ll i = 1;i < n;i++){
auto it = s.fun(f[i].dm);
if(it.first!=-1&&it.first >= f[i].d&&it.first <= f[i].dm){
ans[f[i].pos] = it.second;
s.Remove(it);
}
else {
flag = 0;
break;
}
}
if (flag == 0)puts("No");
else {
puts("Yes");
for (ll i = 1;i < n;i++)
cout << ans[i] << " ";
}
return 0;
}