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;}