题意:一个序列初始为{1},每次复制整个序列添加到原序列后面,并在两个序列间再添个0,所有新增的数字加1。求前M项和。
思路:官方题解是数位DP(玄学)。
自己打了张表找规律,发现对 2^x - 1,其和为 sum(2^(x-1) -1) * 2 + 2 ^ (x-1)。为了方便对于前M项和用sum(M+1)表示。注意到每次复制后项数为2 ^ x - 1。显然是和二进制有关。对于一个被复制的数,是第几次复制的显然就加几。那么我们先将M+1转化成二进制,如100110,然后从左往右找二进制位为1的,对应为2^i。那么产生的和就位sum(2 ^ i) + cnt * 2 ^ i,然后cnt加1,表示后面的是原数字复制cnt + 1次的来的。
复杂度O(M)。
#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>
using namespace std;
typedef long long LL;
#define mem(a, n) memset(a, n, sizeof(a))
#define ALL(v) v.begin(), v.end()
#define si(a) scanf("%d", &a)
#define sii(a, b) scanf("%d%d", &a, &b)
#define siii(a, b, c) scanf("%d%d%d", &a, &b, &c)
#define pb push_back
#define eps 1e-8
const int inf = 0x3f3f3f3f, N = 100 + 5, MOD = 1e9 + 7;
int T, cas = 0;
int n;
LL m, p[60];
int a[N], b[N], tot;
LL sum(LL x) {
if(x == 1) return 0;
if(x == 2) return 1;
return sum(x / 2) * 2 + x / 2;
}
void ex(LL x) {
tot = 0;
while(x) {
a[tot++] = x % 2;
x /= 2;
}
}
void init() {
p[0] = 1;
for(int i = 1; i < 60; i ++) p[i] = p[i-1] * 2;
}
LL work(LL x) {
ex(x);
LL cnt = 0, ans = 0;
for(int i = tot - 1; i >= 0; i --) {
if(a[i] == 1) {
ans += sum(p[i]) + cnt * p[i];
cnt ++;
}
}
return ans;
}
int main(){
#ifdef LOCAL
freopen("/Users/apple/input.txt", "r", stdin);
// freopen("/Users/apple/out.txt", "w", stdout);
#endif
init();
si(T);
while(T --) {
scanf("%lld", &m);
printf("%lld\n", work(m + 1));
}
return 0;
}