Codeforces Round #225 (Div. 1) C 树状数组 || 线段树

来源:互联网 发布:钟恩淇的淘宝店怎么样 编辑:程序博客网 时间:2024/06/06 01:10
看到这题很开心啊,有印象跟以前做过的很像,貌似最近就做过一个,以时间戳为区间来建立树状数组,然后一开始我以为题意是,给x点加val,它以下的所有节点都加-val;所以一开始就以 加 和 减 建立了两个树状数组,最后 减去就是答案,写完发现跟案例对不上啊,读了题目也没发现读错了,对于那句话 我理解错了,后来看了 这个:
http://blog.csdn.net/keshuai19940722/article/details/18967661
仔细看看处理部分,我还以为分奇偶性有规律呢,后来才发现读错题目了,原来是x点加val,与它直接相连的子节点加上-val,它的子节点的子节点又加上val,以此类推。。。哈哈。。哭

跟以前那类题目做法相同,对于这棵树,进行dfs,同时记录当前层数距离跟的关系,用奇偶数来表示,然后再以各个节点被dfs的时间戳 来建立区间 让树状数组映射上去,最后奇偶分开,加的在一个树状数组里,减去的在另一个里面,然后 最后求单点值的时候 就是自己这个点 所属的 距离根节点的关系,也就是自己应该加上的值,再减去对应的另一个树状数组里的应该减去的值,然后 一开始 各个节点本身具有的值 并没有加进树状数组里,还得加上原本具有的值,这样就是答案了

然后又用线段树做了一下,也是以时间戳来搞,同时记录这个节点距离根的奇偶性,然后也是建立两颗线段树,一个记录奇数处理,一个记录偶数处理,结果不知哪里写错了,又改了很久,不行又重新写了一下,真是学啥忘啥。。。

树状数组的:

int n;int m;int c[2][200000 * 2 + 55];typedef struct Node {int l,r,val;int now;};Node node[200000 + 55];vector<int > G[200000 + 55];int cnt;void init() {memset(c,0,sizeof(c));for(int i=0;i<200000 + 55;i++)G[i].clear();}bool input() {while(cin>>n>>m) {for(int i=1;i<=n;i++)cin>>node[i].val;int tmp = n - 1;while(tmp--) {int u,v;scanf("%d %d",&u,&v);G[u].push_back(v);G[v].push_back(u);}return false;}return true;}int lowbit(int x) {return x&(-x);}void add(int i,int val,int *aa) {while(i <= 2 * n) {aa[i] += val;i += lowbit(i);}}int get_sum(int i,int *aa) {int sum = 0;while(i > 0) {sum += aa[i];i -= lowbit(i);}return sum;}void dfs(int u,int pre,int tot) {node[u].l = cnt++;node[u].now = tot;for(int i=0;i<G[u].size();i++) {int v = G[u][i];if(v == pre)continue;dfs(v,u,tot^1);}node[u].r = cnt++;}void cal() {cnt = 1;dfs(1,-1,0);while(m--) {int type;cin>>type;if(type == 1) {int x,y;cin>>x>>y;//int tmp = node[x].now;//int aa = node[x].l;//int bb = node[x].r;add(node[x].l,y,c[node[x].now]);add(node[x].r + 1,-y,c[node[x].now]);}else {int x;cin>>x;//int aa = (get_sum(node[x].l,c[node[x].d]) /*- get_sum(node[x].l - 1,c[node[x].d])*/);//int bb = (get_sum(node[x].l,c[node[x].d^1])/* - get_sum(node[x].l - 1,c[node[x].d^1])*/);//int cc = 0;int ans = get_sum(node[x].l,c[node[x].now]) - get_sum(node[x].l,c[node[x].now^1]);ans += node[x].val;cout<<ans<<endl;}}}void output() {}int main() {while(true) {init();if(input())return 0;cal();output();}return 0;}


线段树的:

const int N = 200000 + 55;int n;int m;int nnum[N + 55];int le[N + 55],ri[N + 55],belong[N + 55];int head[N + 55];typedef struct Node {int l,r;ll sum;int lazy;};Node tree_even[N * 4 + 55],tree_odd[N * 4 + 55];typedef struct NODE {int fro,to;int nex;};NODE edge[2 * N + 55];int tot;int cnt;void add(int u,int v) {edge[tot].fro = u;edge[tot].to = v;edge[tot].nex = head[u];head[u] = tot++;}void dfs(int u,int pre,int d) {le[u] = ++cnt;for(int i=head[u];i!=-1;i=edge[i].nex) {int v = edge[i].to;if(v == pre)continue;dfs(v,u,d^1);}belong[le[u]] = d;ri[le[u]] = cnt;}void push_up(int id,Node *tree) {tree[id].sum = tree[id<<1].sum + tree[id<<1|1].sum;}void push_down(int id,Node *tree) {if(tree[id].lazy == 0)return ;tree[id].sum += (tree[id].r - tree[id].l + 1) * tree[id].lazy;if(tree[id].l == tree[id].r) {tree[id].lazy = 0;return ;}tree[id<<1].lazy += tree[id].lazy;tree[id<<1|1].lazy += tree[id].lazy;tree[id].lazy = 0;}void build(int l,int r,int id,Node *tree) {tree[id].l = l;tree[id].r = r;tree[id].lazy = 0;if(l == r) {tree[id].sum = 0ll;return ;}int mid = (l + r)>>1;build(l,mid,id<<1,tree);build(mid + 1,r,id<<1|1,tree);push_up(id,tree);}void update(int l,int r,int id,ll val,Node *tree) {if(tree[id].l == l && tree[id].r == r) {tree[id].lazy += val;push_down(id,tree);return ;}push_down(id,tree);int mid = (tree[id].l + tree[id].r)>>1;if(r <= mid)update(l,r,id<<1,val,tree);else if(l > mid)update(l,r,id<<1|1,val,tree);else {update(l,mid,id<<1,val,tree);update(mid + 1,r,id<<1|1,val,tree);}push_up(id,tree);}ll query(int l,int r,int id,Node *tree) {if(tree[id].l == l && tree[id].r == r) {push_down(id,tree);return tree[id].sum;}push_down(id,tree);int mid = (tree[id].l + tree[id].r)>>1;ll ret = 0ll;if(r <= mid)ret += query(l,r,id<<1,tree);else if(l > mid)ret += query(l,r,id<<1|1,tree);else {ret += query(l,mid,id<<1,tree);ret += query(mid + 1,r,id<<1|1,tree);}return ret;}void init() {memset(tree_even,0,sizeof(tree_even));memset(tree_odd,0,sizeof(tree_odd));memset(head,-1,sizeof(head));tot = 1;cnt = 0;}bool input() {while(cin>>n>>m) {for(int i=1;i<=n;i++)cin>>nnum[i];for(int i=1;i<n;i++) {int u,v;cin>>u>>v;add(u,v);add(v,u);}return false;}return true;}void cal() {dfs(1,-1,1);build(1,n,1,tree_even);build(1,n,1,tree_odd);while(m--) {int type;cin>>type;if(type == 1) {int x,y;cin>>x>>y;int left = le[x];int right = ri[left];if(belong[left]&1) update(left,right,1,y,tree_odd);else update(left,right,1,y,tree_even);}else {int x;cin>>x;int left = le[x];ll ans;if(belong[left]&1)ans = query(left,left,1,tree_odd) - query(left,left,1,tree_even);elseans = query(left,left,1,tree_even) - query(left,left,1,tree_odd);ans += nnum[x];cout<<ans<<endl;}}}void output() {}int main() {while(true) {init();if(input())return 0;cal();output();}return 0;}


0 0
原创粉丝点击