2019杭电多校第九场 1002 Rikka with Cake(树状数组+离散化)

题意:在一个矩形中有若干条平行坐标轴的直线 ,求将矩形分成多少块。

分析:根据离散数学公式可以推出分成的区域就是k条直线的交点数+1,将问题转化
先将坐标离散化 然后对于其中为L方向的点加入树状数组, 依次枚举每各点的坐标
如果方向向上 则查询树状数组后缀和 若向下 查询前缀和
若向左则将树状数组的这个点删去 因为此后他不在贡献 若向右再加入树状数组 贡献一直到结尾位置

//http://acm.hdu.edu.cn/showproblem.php?pid=6681
#include
using namespace std;
typedef long long ll;
const int maxn = 4e5 + 5;
int c[maxn];
int n, m, k;
inline int lowbit(int x){ return x&-x; }
inline void add(int x, int y){
	while (x <= m){
		c[x] += y;
		x += lowbit(x);
	}
}
inline int query(int x){
	int res = 0;
	while (x > 0){
		res += c[x];
		x -= lowbit(x);
	}
	return res;
}
struct node{
	int x, y;
	char ch;
	bool operator<(const node &a)const{
		return x < a.x;
	}
}a[maxn];
vector<int>vx, vy;
int getidx(int x){ return lower_bound(vx.begin(), vx.end(), x) - vx.begin(); }
int getidy(int x){ return lower_bound(vy.begin(), vy.end(), x) - vy.begin(); }
int main(){
	int T;
	cin >> T;
	while (T--){
		memset(c, 0, sizeof c);
		vx.clear(); vy.clear();
		scanf("%d%d%d", &n, &m, &k);
		for (int i = 1; i <= k; i++){
			scanf("%d%d %c", &a[i].x, &a[i].y, &a[i].ch);
			vx.push_back(a[i].x), vy.push_back(a[i].y);
		}
		sort(a + 1, a + 1 + k);
		//离散化坐标
		vx.push_back(INT_MIN), vy.push_back(INT_MIN);
		sort(vx.begin(), vx.end());
		vx.erase(unique(vx.begin(), vx.end()), vx.end());
		sort(vy.begin(), vy.end());
		vy.erase(unique(vy.begin(), vy.end()), vy.end());
		m = vy.size() + 1;//纵坐标的最大值  即树状数组的范围

		for (int i = 1; i <= k; i++)
			if (a[i].ch == 'L'){
				int x = getidx(a[i].x), y = getidy(a[i].y);
				add(y, 1);
			}

		ll ans = 0;
		for (int i = 1; i <= k; i++){
            int x = getidx(a[i].x), y = getidy(a[i].y);
			if (a[i].ch == 'U')
				ans += query(m) - query(y - 1);
			else if (a[i].ch == 'L')
				add(y, -1);
			else if (a[i].ch == 'R')
				add(y, 1);
			else if (a[i].ch == 'D')
				ans += query(y);
		}
		printf("%lld\n", ans+1);
	}
	return 0;
}

你可能感兴趣的:(HDU,数据结构)