题意:
有一棵树,给出两种操作分别为子树值修改与节点值查询操作。
解法:
设根节点为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,应该是编译器的问题额。
有一棵树,给出两种操作分别为子树值修改与节点值查询操作。
解法:
设根节点为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;
}