UVa12532 - Interval Product(Fenwick树)

最新推荐文章于 2015-08-08 09:33:00 发布

原创最新推荐文章于 2015-08-08 09:33:00 发布 · 896 阅读

0 ·

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

OJ 同时被 2 个专栏收录

986 篇文章

订阅专栏

算法设计与分析

321 篇文章

订阅专栏

本文介绍了一个基于Fenwick树（二叉索引树）实现的区间查询与更新操作的数据结构应用案例。该程序能够高效处理数组中元素的查询与修改，并支持查询指定区间内的元素个数。特别地，针对负数和零值进行了专门处理。

#include <cstdio>
#include <vector>

using namespace std;

const int N = 100010;

struct Fenwick
{
private:
	vector<int> fw;
	
public:
	int lowbit(int n)
	{
		return n & (-n);
	}
	
	void init(int n)
	{
		fw.assign(n + 1, 0);
	}
	
	int sum(int n)
	{
		int s = 0;
		while (n > 0) {
			s += fw[n];
			n -= lowbit(n);
		}
		
		return s;
	}
	
	int sum(int a, int b)
	{
		return sum(b) - (a == 1 ? 0 : sum(a - 1));
	}
	
	void adjust(int k, int v)
	{
		while (k < (int)fw.size()) {
			fw[k] += v;
			k += lowbit(k);
		}
	}
};

struct solution
{
private:
	int x[N];
	int n, k;
	Fenwick neg, zero;
	
public:
	bool input()
	{
		if (scanf("%d%d", &n, &k) != 2) return false;
		
		neg.init(n);
		zero.init(n);
		
		for (int i = 1; i <= n; i++) {
			scanf("%d", &x[i]);
			if (x[i] < 0) neg.adjust(i, 1);
			else if (x[i] == 0) zero.adjust(i, 1);
		}
		
		return true;
	}
	
	void solve()
	{
		for (int i = 0; i < k; i++) {
			char ch;
			scanf("%c", &ch);

			while (ch != 'P' && ch != 'C') {
				scanf("%c",  &ch);
			}
			
			int a, b;
			scanf("%d%d", &a, &b);
			
			if (ch == 'P') {
				if (zero.sum(a, b)) {
					printf("0");
				} else if (neg.sum(a, b) & 1) {
					printf("-");
				} else {
					printf("+");
				}
			} else {
				if (x[a] < 0 && b >= 0) {
					neg.adjust(a, 1);
				} else if (x[a] >= 0 && b < 0) {
					neg.adjust(a, 1);
				}
				
				if (x[a] != 0 && b == 0) {
					zero.adjust(a, 1);
				} else if (x[a] == 0 && b != 0) {
					zero.adjust(a, -1);
				}
				
				x[a] = b;
			}
		}
		
		printf("\n");
	}
};

int main()
{
#ifndef ONLINE_JUDGE
	freopen("d:\\OJ\\uva_in.txt", "r", stdin);
#endif

	solution solver;
	while (solver.input()) {
		solver.solve();
	}
	 
	return 0;
}