P2038 无线网络发射器选址

最新推荐文章于 2025-04-18 21:12:21 发布

原创最新推荐文章于 2025-04-18 21:12:21 发布 · 235 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

洛谷题解专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种使用二维树状数组解决区间和问题的方法。通过将一维问题转化为二维，文章详细展示了如何利用树状数组进行高效的数据更新和查询操作。代码示例清晰地解释了lowbit函数的运用，以及如何在正方形边界内正确计算区间和，避免边界问题。通过具体实例，文章演示了如何初始化树状数组，进行元素添加和区间查询，最终找出最大计数及相应计数的个数。

这道题我用来练习二维树状数组
把一维变成二维思路也很简单，代码不麻烦
关键是计算区间和时要注意正方形边界是否包含，是否-1

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>

inline int lowbit(int x)
{
	return (x & (-x));
}
using namespace std;

int d, n,x,y,k;
int tree[130][130];
inline int read()
{
	int x = 0, y = 1; char c = getchar();
	while (c>'9' || c<'0'){ if (c == '0')y = -1; c = getchar(); }
	while (c >= '0'&&c <= '9'){ x = x * 10 + c - '0'; c = getchar(); }
	return x*y;
}

void add(int x, int y, int k)
{
	for (int i = x; i <= 129; i += lowbit(i))
	{
		for (int j = y; j <= 129; j += lowbit(j))
		{
			tree[i][j] += k;
		}
	}
}

int query(int x, int y)
{
	int ans = 0;
	for (int i = x; i >= 1; i -= lowbit(i))
	{
		for (int j = y; j >= 1; j -= lowbit(j))
		{
			ans += tree[i][j];
		}
	}
	return ans;
}
int main()
{
	d = read();
	n = read();
	for (int i = 0; i < n; ++i)
	{
		x = read();
		y = read();
		k = read();
		add(x+1, y+1, k);
	}

	int maxcount = 0;
	int anscount = 0;
	for (int i = 1; i <= 129 ; ++i)
	{
		for (int j = 1; j <= 129 ; ++j)
		{
			int x2 = i + d;
			if (x2 > 129)x2 = 129;
			int y2 = j + d;
			if (y2 > 129)y2 = 129;
			int x1 = i - d;
			if (x1<1)x1 = 1;
			int y1 = j - d;
			if (y1 <1)y1 = 1;
			int count = query(x2, y2) - query(x2,y1-1)
				- query(x1-1,y2) + query(x1-1,y1-1);//注意边界
			if (count > maxcount)
			{
				maxcount = count;
				anscount = 1;
			}
			else if (count == maxcount)
			{
				anscount++;
			}
		}
	}
	cout << anscount <<" " << maxcount;
	return 0;
}