UVA 1406 - A Sequence of Numbers(树状数组)

UVA 1406 - A Sequence of Numbers

题目链接

题意:给定一些数字,每次操作
C x 表示所有数字加上x
Q x 表示答案加上与2x进行且操作不为0的个数
E 结束,并输出答案

思路:树状数组,首先观察,对于每次查询x而言,只有前x位是是有用的,所以可以开16个树状数组,每个数组表示数字对应前x位下的数字,然后就可以搞了,已经现在加过值为sum,可以推出
2x<=sum+num<2x,这样就可以根据sum来推出num的相应区间,利用区间查询进行查找

代码:

#include <cstdio>
#include <cstring>

#define lowbit(x) (x&(-x))
const int N = 66666;

typedef long long ll;

int n, bit[16][N], mi[20];

void add(int i, int x, int v) {
    while (x < N) {
	bit[i][x] += v;
	x += lowbit(x);
    }
}

int get(int i, int x) {
    int ans = 0;
    while (x) {
	ans += bit[i][x];
	x -= lowbit(x);
    }
    return ans;
}

void init() {
    memset(bit, 0, sizeof(bit));
    int num;
    while (n--) {
	scanf("%d", &num);
	for (int i = 1; i <= 16; i++) {
	    add(i - 1, num % mi[i] + 1, 1);
	}
    }
}

ll solve() {
    char op[2];
    int d, sum = 0;
    ll ans = 0;
    while (scanf("%s", op)) {
	if (op[0] == 'E') break;
	scanf("%d", &d);
	if (op[0] == 'C')
	    sum = (sum + d) % mi[16];
	else {
	    int tmp = sum % mi[d];
	    if (sum&mi[d])
		ans += get(d, mi[d] - tmp) + get(d, mi[d + 1]) - get(d, mi[d + 1] - tmp);
	    else ans += get(d, mi[d + 1] - tmp) - get(d, mi[d] - tmp);
	}
    }
    return ans;
}

int main() {
    int cas = 0;
    mi[0] = 1;
    for (int i = 1; i <= 16; i++)
	mi[i] = mi[i - 1] * 2;
    while (~scanf("%d", &n) && n != -1) {
	init();
	printf("Case %d: %lld\n", ++cas, solve());
    }
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值