【bzoj1604】[Usaco2008 Open]Cow Neighborhoods 奶牛的邻居 旋转坐标系+并查集+Treap/STL-set

题目描述

了解奶牛们的人都知道,奶牛喜欢成群结队.观察约翰的N(1≤N≤100000)只奶牛,你会发现她们已经结成了几个“群”.每只奶牛在吃草的时候有一个独一无二的位置坐标Xi,Yi(l≤Xi,Yi≤[1..10^9];Xi,Yi∈整数.当满足下列两个条件之一,两只奶牛i和j是属于同一个群的:
1.两只奶牛的曼哈顿距离不超过C(1≤C≤10^9),即lXi - xil+IYi - Yil≤C.
2.两只奶牛有共同的邻居.即,存在一只奶牛k,使i与k,j与k均同属一个群.
给出奶牛们的位置,请计算草原上有多少个牛群,以及最大的牛群里有多少奶牛

输入

1行输入N和C,之后N行每行输入一只奶牛的坐标.

输出

仅一行,先输出牛群数,再输出最大牛群里的牛数,用空格隔开.

样例输入

4 2
1 1
3 3
2 2
10 10

样例输出

2 3


题解

为了练习Treap找到的这道略神的题

首先直接处理曼哈顿距离不是特别容易,我们可以把所有的点绕着原点逆时针旋转45°,这样原来的点$(x,y)$就变为了$(frac{x-y}{sqrt 2},frac{x+y}{sqrt 2})$,查询的区域变为了矩形范围,切比雪夫距离(横纵坐标差的绝对值最大值)不超过$frac c{sqrt 2}$。

然后约掉$frac 1{sqrt 2}$,就变为普通的矩形区域查询问题。

先将所有变换后的点按照横坐标排序,然后从左往右扫,将左面横坐标不满足条件的点删除。然后考虑连边:我们没有必要将所有在范围之内的点与当前点连边,只需要将当前点与第一个纵坐标比它大的点、第一个纵坐标比它小的点,如果满足条件就连边。

证明:使用数学归纳法

两个点之间使用这种方法是一定能够连上的。

如果k个点连上了,且纵坐标都比当前点大,并且横坐标满足条件,如果这种方法是不成立的,那么不妨设y1、y2,其中y1为纵坐标最接近当前点,y2为要连的点,我们要证的就是“当前点与y2有边,与y1没有边”是假命题。证明显然~

纵坐标比当前点小的时候同理。

于是k+1个点也能连上。命题得证。

回到题中,删点加点、查询前驱后继可以使用平衡树,维护连通性可以使用并查集。最后扫一遍每个点即可得到答案。

时间复杂度$O(nlog n)$。

事实上,STL的set比Treap还快~

Treap:

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define N 100010
using namespace std;
struct data
{
	int x , y;
}a[N];
typedef pair<int , int> pr;
int l[N] , r[N] , rnd[N] , tot , root , f[N] , tmp , num[N];
pr w[N];
bool cmp(data a , data b)
{
	return a.x < b.x;
}
void zig(int &k)
{
	int t = l[k];
	l[k] = r[t] , r[t] = k , k = t;
}
void zag(int &k)
{
	int t = r[k];
	r[k] = l[t] , l[t] = k , k = t;
}
void insert(int &k , pr x)
{
	if(!k) k = ++tot , w[k] = x , rnd[k] = rand();
	else if(x < w[k])
	{
		insert(l[k] , x);
		if(rnd[l[k]] < rnd[k]) zig(k);
	}
	else
	{
		insert(r[k] , x);
		if(rnd[r[k]] < rnd[k]) zag(k);
	}
}
void del(int &k , pr x)
{
	if(x == w[k])
	{
		if(!l[k] || !r[k]) k = l[k] + r[k];
		else if(rnd[l[k]] < rnd[k]) zig(k) , del(r[k] , x);
		else zag(k) , del(l[k] , x);
	}
	else if(x < w[k]) del(l[k] , x);
	else del(r[k] , x);
}
void pre(int k , pr x)
{
	if(!k) return;
	else if(x < w[k]) pre(l[k] , x);
	else tmp = w[k].second , pre(r[k] , x);
}
void sub(int k , pr x)
{
	if(!k) return;
	else if(x < w[k]) tmp = w[k].second , sub(l[k] , x);
	else sub(r[k] , x);
}
int find(int x)
{
	return x == f[x] ? x : f[x] = find(f[x]);
}
int main()
{
	int n , c , i , u , v , p = 1 , ans = 0 , mx = 0;
	scanf("%d%d" , &n , &c);
	for(i = 1 ; i <= n ; i ++ ) scanf("%d%d" , &u , &v) , a[i].x = u - v , a[i].y = u + v , f[i] = i;
	sort(a + 1 , a + n + 1 , cmp);
	for(i = 1 ; i <= n ; i ++ )
	{
		while(p < i && a[i].x - a[p].x > c) del(root , pr(a[p].y , p)) , p ++ ;
		tmp = 0 , pre(root , pr(a[i].y , i));
		if(tmp && a[i].y - a[tmp].y <= c) f[find(i)] = find(tmp);
		tmp = 0 , sub(root , pr(a[i].y , i));
		if(tmp && a[tmp].y - a[i].y <= c) f[find(i)] = find(tmp);
		insert(root , pr(a[i].y , i));
	}
	for(i = 1 ; i <= n ; i ++ ) num[find(i)] ++ ;
	for(i = 1 ; i <= n ; i ++ )
		if(num[i])
			ans ++ , mx = max(mx , num[i]);
	printf("%d %d
" , ans , mx);
	return 0;
}

STL-set:

#include <cstdio>
#include <algorithm>
#include <set>
#define N 100010
using namespace std;
struct data
{
    int x , y;
}a[N];
typedef pair<int , int> pr;
set<pr> s;
set<pr>::iterator it;
int f[N] , num[N];
bool cmp(data a , data b)
{
    return a.x < b.x;
}
int find(int x)
{
    return x == f[x] ? x : f[x] = find(f[x]);
}
int main()
{
    int n , c , i , u , v , p = 1 , ans = 0 , mx = 0;
    scanf("%d%d" , &n , &c);
    for(i = 1 ; i <= n ; i ++ ) scanf("%d%d" , &u , &v) , a[i].x = u - v , a[i].y = u + v , f[i] = i;
    sort(a + 1 , a + n + 1 , cmp);
    for(i = 1 ; i <= n ; i ++ )
    {
        while(p < i && a[i].x - a[p].x > c) s.erase(pr(a[p].y , p)) , p ++ ;
        it = s.upper_bound(pr(a[i].y , i));
        if(it != s.end() && it->first - a[i].y <= c) f[find(i)] = find(it->second);
        if(it != s.begin() && a[i].y - (--it)->first <= c) f[find(i)] = find(it->second);
        s.insert(pr(a[i].y , i));
    }
    for(i = 1 ; i <= n ; i ++ ) num[find(i)] ++ ;
    for(i = 1 ; i <= n ; i ++ )
        if(num[i])
            ans ++ , mx = max(mx , num[i]);
    printf("%d %d
" , ans , mx);
    return 0;
}

 

原文地址:https://www.cnblogs.com/GXZlegend/p/7148774.html