蓝桥杯 Log大侠 【线段树+“剪枝”】

                                        Log大侠

Description

atm参加了速算训练班，经过刻苦修炼，对以2为底的对数算得飞快，人称Log大侠。
一天，Log大侠的好友 drd 有一些整数序列需要变换，Log大侠正好施展法力...
变换的规则是： 对其某个子序列的每个整数变为: [log_2 (x) + 1]  其中 [] 表示向下取整，
就是对每个数字求以2为底的对数，然后取下整。
例如对序列 3 4 2 操作一次后，这个序列会变成 2 3 2。
drd需要知道，每次这样操作后，序列的和是多少。
【输入格式】
第一行两个正整数 n m 。
第二行 n 个数，表示整数序列，都是正数。
接下来 m 行，每行两个数 L R 表示 atm 这次操作的是区间 [L, R]，数列序号从1开始。
【输出格式】
输出 m 行，依次表示 atm 每做完一个操作后，整个序列的和。

例如，输入：
3 3
5 6 4
1 2
2 3
1 3
程序应该输出：
10
8
6

【数据范围】
对于 30% 的数据， n, m <= 10^3
对于 100% 的数据， n, m <= 10^5
资源约定：
峰值内存消耗 < 256M
CPU消耗  < 1000ms

解题思路
   可以很容易看出来这是个线段树的板子题，但是需要“剪枝”，-，，就姑且称为“剪枝”吧，  假如一个区间内的数全为1或者全为0，那么就没必要再逐个找到这个区间里的数取log2+1了，因为log2（1）+1 = 1，log2(2)+1 = 2

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
struct Node {
	int l,r;
	ll w;
	bool flag;
}tree[4*maxn];
void Build(int l,int r,int k)
{
	tree[k].l = l;
	tree[k].r = r;
	if(tree[k].l == tree[k].r) {
		scanf("%lld",&tree[k].w);
		if(tree[k].w == 1 || tree[k].w == 2)
			tree[k].flag = true;
		return;
	}
	int m = l+(r-l)/2;
	Build(l,m,k*2);
	Build(m+1,r,k*2+1);
	tree[k].w = tree[k*2].w + tree[k*2+1].w;
	tree[k].flag = tree[k*2].flag & tree[k*2+1].flag;
}
void Update(int l,int r,int k)
{
	
	if(tree[k].flag)		// 剪枝 
		return;
	if(tree[k].l == tree[k].r) {
		tree[k].w = log2(tree[k].w)+1;
		if(tree[k].w ==1 || tree[k].w == 2)
			tree[k].flag = true;
		return ;
	}
	int m = tree[k].l + (tree[k].r-tree[k].l)/2;
	if(l <= m) 
		Update(l,r,k*2);
	if(r > m) 
		Update(l,r,k*2+1);
	tree[k].w = tree[k*2].w + tree[k*2+1].w;
	tree[k].flag = tree[k*2].flag & tree[k*2+1].flag;
}
int main()
{
	int n,m;
	scanf("%d %d",&n,&m);
	Build(1,n,1);
	int l,r;
	while( m--) {
		scanf("%d %d",&l,&r);
		Update(l,r,1);
		printf("%lld\n",tree[1].w);
	}
	return 0; 
}