FZUOJ 2277 Change

题意:
有一棵树,给出两种操作分别为子树值修改与节点值查询操作。
解法:
设根节点为v,子节点为s,对v点的修改,即a[s] += x + (dep[v] - dep[s])*k,
化简得a[s] += x + dep[v]*k - dep[s]*k。采用两个树状数组来进行区间修改,单点查询。
利用dfs序变树为线性结构,进而得到修改区间。

dfs序:l[x]为dfs序中x的位置,r[x]为dfs序中x子树的结束位置。
一颗子树的所有节点在DFS序内是连续的一段, 利用这个性质我们可以解决很多问题。

这题让我无数次爆了int,对精度的控制有了全新的认识。。。
还出现了让我莫名其妙的bug,应该是编译器的问题额。
#ifdef _DEBUG
#pragma warning(disable : 4996)
#endif
#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <algorithm>
#include <functional>
#include <sstream>
#include <utility>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <cmath>
#include <cctype>
#define CLEAR(a, b) memset(a, b, sizeof(a))
#define CLOSE() ios::sync_with_stdio(false)
#define IN() freopen("in.txt", "r", stdin)
#define OUT() freopen("out.txt", "w", stdout)
#define PF(a) printf("%d\n", a)
#define SF(a) scanf("%d", &a)
#define SFF(a, b) scanf("%d%d", &a, &b)
#define SFFF(a, b, c) scanf("%d%d%d", &a, &b, &c)
#define FOR(i, a, b) for(int i = a; i < b; ++i)
#define LL long long
#define maxn 300005
#define maxm 1000005
#define mod  1000000007
#define INF 10000000
#define EPS 1e-6
#define N 26
using namespace std;
//-------------------------CHC------------------------------//
#define lbt(x) x & -x
vector<int> v[maxn];
int n, t, l[maxn], r[maxn], dep[maxn];
LL bit1[maxn], bit2[maxn];

void update(LL *bit, int idx, LL val) {
	while (idx <= n) {
		bit[idx] = ((bit[idx] + val) % mod + mod) % mod;
		idx += lbt(idx);
	}
}

int sum(int l, int v) {
	LL ret1 = 0, ret2 = 0;
	for (int i = l; i; i -= lbt(i)) ret1 = (ret1 + bit1[i]) % mod;
	for (int i = l; i; i -= lbt(i)) ret2 = (ret2 + bit2[i]) % mod;
	return ((ret1 - ret2*dep[v]) % mod + mod) % mod;
}

void dfs(int s, int depth) {
	l[s] = ++t;
	dep[s] = depth;
	FOR(i, 0, v[s].size()) dfs(v[s][i], depth + 1);
	r[s] = t;
}

int main() {
	int T, q;
	SF(T);
	while (T--) {
		CLEAR(bit1, 0);
		CLEAR(bit2, 0);
		SF(n);
		FOR(i, 1, n + 1) v[i].clear();
		int u;
		FOR(i, 2, n + 1) SF(u), v[u].push_back(i);
		t = 0;
		dfs(1, 1);
		SF(q);
		int op, v, x, k;
		while (q--) {
			SFF(op, v);
			if (op == 1) {
				SFF(x, k);
				LL ret = x + dep[v] * (LL)k;
				update(bit1, l[v], ret);
				update(bit1, r[v] + 1, -ret);
				update(bit2, l[v], k);
				update(bit2, r[v] + 1, -k);
			}
			else PF(sum(l[v], v));
		}
	}
	return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值