题目描述
输入
输出
样例输入
10 7
1 1 1 1
1 2 2 3
1 4 1 2
1 3 4 4
2 1 1 4 1 3
2 2 2 3 5 4
2 2 1 4 4 2
样例输出
NAIVE!ORZzyz.
NAIVE!ORZzyz.
3
题解
权值线段树套KD-tree
对于外层权值线段树的每个节点,对应着一棵KD-tree,存储权值在外层节点范围内的所有点。
对于每次询问,如果判定有解,则寻找区间右半部分中点的个数,如果大于等于k则在右区间中找,否则在左区间中找,直至l=r得到答案。
嗯,说起来真是容易。然而这样交上去肯定是必T无疑。
究其原因就是在KD-tree上。
一开始把“平衡”KD-tree的时间复杂度当成$O(\log n)$的了,以为稳过,结果卡得很死。
由于KD-tree不能旋转什么的,所以自然不能保证平衡。
那么我们能做的只有对KD-tree进行重构,但是蒟蒻又不会部分重构(子树大小超过某比例时重构,替罪羊树的操作),所以只能固定点数重构。
然而重构又出了各种各样的问题 = =,改了以后交上去还是TLE。
实在没办法,要了份数据,发现第6、7个点卡不重构的KD-tree,第8、9、10个点数据范围非常大,这导致无论怎么改重构时间都无法通过。
最后只能针对数据来解决问题了(逃),m=50000时是前7个点,m=100000时是后3个点,分情况就好了,最后还是勉强跑过了。
说实话真正考试如果出这种题的话80分真的是满足了。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 3000010
using namespace std;
const int M = 1000000000;
int m , R , ls[N] , rs[N] , root[N] , ts[N] , tot , d , num , x1 , y1 , x2 , y2;
int sta[N] , top;
struct data
{
int p[2] , mx[2] , mn[2] , si , c[2];
bool operator<(const data a)const {return p[d] == a.p[d] ? p[d ^ 1] < a.p[d ^ 1] : p[d] < a.p[d];}
}a[N];
bool cmp(int x , int y)
{
return a[x].p[d] == a[y].p[d] ? a[x].p[d ^ 1] < a[y].p[d ^ 1] : a[x].p[d] < a[y].p[d];
}
void pushup(int k)
{
int l = a[k].c[0] , r = a[k].c[1];
a[k].mx[0] = max(a[k].p[0] , max(a[l].mx[0] , a[r].mx[0]));
a[k].mx[1] = max(a[k].p[1] , max(a[l].mx[1] , a[r].mx[1]));
a[k].mn[0] = min(a[k].p[0] , min(a[l].mn[0] , a[r].mn[0]));
a[k].mn[1] = min(a[k].p[1] , min(a[l].mn[1] , a[r].mn[1]));
a[k].si = a[l].si + a[r].si + 1;
}
void insert(int &k)
{
if(!k) k = ++num , a[k].p[0] = x1 , a[k].p[1] = y1;
else if((d == 0 && (x1 == a[k].p[0] ? y1 < a[k].p[1] : x1 < a[k].p[0])) || (d == 1 && (y1 == a[k].p[1] ? x1 < a[k].p[0] : y1 < a[k].p[1]))) d ^= 1 , insert(a[k].c[0]);
else insert(a[k].c[1]);
pushup(k);
}
int query(int k)
{
if(!k || x1 > a[k].mx[0] || y1 > a[k].mx[1] || x2 < a[k].mn[0] || y2 < a[k].mn[1]) return 0;
if(x1 <= a[k].mn[0] && y1 <= a[k].mn[1] && x2 >= a[k].mx[0] && y2 >= a[k].mx[1]) return a[k].si;
int ans = (a[k].p[0] >= x1 && a[k].p[1] >= y1 && a[k].p[0] <= x2 && a[k].p[1] <= y2);
ans += query(a[k].c[0]);
ans += query(a[k].c[1]);
return ans;
}
int solve(int k , int l , int r , int x)
{
if(l == r) return l;
int mid = (l + r) >> 1 , sum = query(root[rs[x]]);
if(sum >= k) return solve(k , mid + 1 , r , rs[x]);
else return solve(k - sum , l , mid , ls[x]);
}
int build(int l , int r , int now)
{
if(l > r) return 0;
int mid = (l + r) >> 1 , pos;
d = now , nth_element(sta + l , sta + mid , sta + r + 1 , cmp) , pos = sta[mid];
a[pos].c[0] = build(l , mid - 1 , now ^ 1);
a[pos].c[1] = build(mid + 1 , r , now ^ 1);
pushup(pos);
return pos;
}
void dfs(int k)
{
if(!k) return;
sta[++top] = k;
dfs(a[k].c[0]) , dfs(a[k].c[1]);
}
void rebuild(int &k)
{
top = 0 , dfs(k);
k = build(1 , top , 0);
}
void add(int p , int l , int r , int &x)
{
if(!x) x = ++tot;
d = 0 , insert(root[x]) , ts[x] ++ ;
if(m == 50000 && ts[x] >= 2000) rebuild(root[x]) , ts[x] = 0;
if(l == r) return;
int mid = (l + r) >> 1;
if(p <= mid) add(p , l , mid , ls[x]);
else add(p , mid + 1 , r , rs[x]);
}
int main()
{
a[0].mx[0] = a[0].mx[1] = -M , a[0].mn[0] = a[0].mn[1] = M;
int ans = 0 , opt , v , i;
scanf("%*d%d" , &m);
for(i = 1 ; i <= m ; i ++ )
{
scanf("%d" , &opt);
if(opt == 1)
{
scanf("%d%d%d" , &x1 , &y1 , &v);
x1 ^= ans , y1 ^= ans , v ^= ans;
add(v , 1 , M , R);
}
else
{
scanf("%d%d%d%d%d" , &x1 , &y1 , &x2 , &y2 , &v);
x1 ^= ans , y1 ^= ans , x2 ^= ans , y2 ^= ans , v ^= ans;
if(query(root[R]) < v) puts("NAIVE!ORZzyz.") , ans = 0;
else printf("%d\n" , ans = solve(v , 1 , M , R));
}
}
return 0;
}
17.12.18 UPD
写了一下替罪羊式重构的写法,好像也不是很难嘛。。。
在KD-tree插入过程中直接判断一个点是否失去平衡,是的话就直接重构即可。方法和替罪羊树类似,可以参考替罪羊树例题 【bzoj3065】带插入区间k小值 。
并且稍微优化了一下代码的细节,现在差不多可读了。。。
时间复杂度终于是稳定的 $O(n\sqrt n\log n)$ 了,终于不用针对数据啦。
#include <cstdio>
#include <algorithm>
#define N 100010
#define M 3000010
using namespace std;
struct seg
{
int ls , rs , rb;
}a[M];
struct kdt
{
int p[2] , mx[2] , mn[2] , ls , rs , si;
}b[M];
int A[2] , B[2] , d , root , id[N] , ta , tb , tot;
inline bool tc(int *a , int *b)
{
return a[d] == b[d] ? a[d ^ 1] < b[d ^ 1] : a[d] < b[d];
}
bool cmp(int x , int y)
{
return tc(b[x].p , b[y].p);
}
inline void pushup(int x)
{
int l = b[x].ls , r = b[x].rs;
b[x].mx[0] = max(b[x].p[0] , max(b[l].mx[0] , b[r].mx[0]));
b[x].mx[1] = max(b[x].p[1] , max(b[l].mx[1] , b[r].mx[1]));
b[x].mn[0] = min(b[x].p[0] , min(b[l].mn[0] , b[r].mn[0]));
b[x].mn[1] = min(b[x].p[1] , min(b[l].mn[1] , b[r].mn[1]));
b[x].si = b[l].si + b[r].si + 1;
}
int build(int l , int r , int now)
{
if(l > r) return 0;
int mid = (l + r) >> 1;
d = now , nth_element(id + l , id + mid , id + r + 1 , cmp);
b[id[mid]].ls = build(l , mid - 1 , now ^ 1);
b[id[mid]].rs = build(mid + 1 , r , now ^ 1);
pushup(id[mid]);
return id[mid];
}
void dfs(int k)
{
if(k) dfs(b[k].ls) , id[++tot] = k , dfs(b[k].rs);
}
void insert(int &k , int now , int flag)
{
if(!k)
{
k = ++tb , b[k].p[0] = A[0] , b[k].p[1] = A[1] , pushup(k);
return;
}
bool tag;
d = now;
if(tc(A , b[k].p)) tag = ((b[b[k].ls].si + 1) * 4 > (b[k].si + 1) * 3) , insert(b[k].ls , now ^ 1 , tag | flag);
else tag = ((b[b[k].rs].si + 1) * 4 > (b[k].si + 1) * 3) , insert(b[k].rs , now ^ 1 , tag | flag);
pushup(k);
if(tag && !flag) tot = 0 , dfs(k) , k = build(1 , tot , now);
}
int query(int k)
{
if(!k || b[k].mx[0] < A[0] || b[k].mx[1] < A[1] || b[k].mn[0] > B[0] || b[k].mn[1] > B[1]) return 0;
if(b[k].mn[0] >= A[0] && b[k].mn[1] >= A[1] && b[k].mx[0] <= B[0] && b[k].mx[1] <= B[1]) return b[k].si;
return (b[k].p[0] >= A[0] && b[k].p[1] >= A[1] && b[k].p[0] <= B[0] && b[k].p[1] <= B[1]) + query(b[k].ls) + query(b[k].rs);
}
void modify(int p , int l , int r , int &x)
{
if(!x) x = ++ta;
insert(a[x].rb , 0 , 0);
if(l == r) return;
int mid = (l + r) >> 1;
if(p <= mid) modify(p , l , mid , a[x].ls);
else modify(p , mid + 1 , r , a[x].rs);
}
int solve(int k , int l , int r , int x)
{
if(l == r) return l;
int mid = (l + r) >> 1 , t = query(a[a[x].rs].rb);
if(k <= t) return solve(k , mid + 1 , r , a[x].rs);
else return solve(k - t , l , mid , a[x].ls);
}
int main()
{
b[0].mx[0] = b[0].mx[1] = -1 << 30 , b[0].mn[0] = b[0].mn[1] = 1 << 30;
int m , opt , x , last = 0;
scanf("%*d%d" , &m);
while(m -- )
{
scanf("%d" , &opt);
if(opt == 1) scanf("%d%d%d" , &A[0] , &A[1] , &x) , A[0] ^= last , A[1] ^= last , x ^= last , modify(x , 0 , 1000000000 , root);
else
{
scanf("%d%d%d%d%d" , &A[0] , &A[1] , &B[0] , &B[1] , &x), A[0] ^= last , A[1] ^= last , B[0] ^= last , B[1] ^= last , x ^= last;
if(!(last = solve(x , 0 , 1000000000 , root))) puts("NAIVE!ORZzyz.");
else printf("%d\n" , last);
}
}
return 0;
}