ZOJ-3279

算一道比较水的线段树,用树状数组应该也可以,但是感觉操作起来比线段树麻烦。我的思路是维护区间的和,然后根据输入的rank由根到底查,可和左儿子比较大小然后决定向左走还是向右走,直到底部输出index,1A了,感觉不错,哈哈

#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

namespace
{
	const int MAX_N = (1 << 18) - 1;
	int n, N, K, q;
	int *dat = new int[MAX_N];

	int init(int index)
	{
		if (index < n - 1)
			return dat[index] = init(2 * index + 1) + init(2 * index + 2);
		else
			return dat[index];
	}

	void input()
	{
		memset(dat, 0, sizeof(int) * MAX_N);
		n = 1;
		while (n < N)
			n <<= 1;
		int index = n - 1;
		for (int i = 0; i < N; i++)
			scanf("%d", &dat[index++]);
		init(0);
	}

	void update(int a, int b)
	{
		int index = a + n - 2;
		dat[index] = b;
		while (index)
		{
			index = (index - 1) >> 1;
			dat[index] = dat[2 * index + 1] + dat[2 * index + 2];
		}
	}

	void solve()
	{
		int index = 0, l;
		while (index < n - 1)
		{
			l = dat[index * 2 + 1];
			if (q > l)
			{
				index = index * 2 + 2;
				q -= l;
			}
			else
				index = index * 2 + 1;
		}
		printf("%d\n", index - n + 2);
	}
}

int main()
{
	char s[2];
	int a, b;
	while (scanf("%d", &N) != EOF)
	{
		input();
		scanf("%d", &K);
		while (K--)
		{
			scanf("%s", s);
			if (s[0] == 'p')
			{
				scanf("%d %d", &a, &b);
				update(a, b);
			}
			else
			{
				scanf("%d", &q);
				solve();
			}
		}
	}
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值