HDU:6682-Rikka with Mista

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6682

题意：现在有n个数，你可以任意组合，问在所有的组合中，对每个组合求和，那么总共含有多少个数字4。

解题心得：

• 看到$n$最大为$40$，第一个反应肯定就是折半搜索，首先将所有的$n$个数均分成两半，并且分别将每一半的所有组合求和暴力枚举出来，设存放在$ve1$和$ve2$中。接下来就可以枚举$ve1$中的每一个数设当前数为$x$，枚举$x$中的每个位数，然后可以直接在另一半中找有多少个可以和$x$加起来变成$4$，如$x$等于$2$时可以在另一边找和$x$十进制位相同时有多少个$2$，第二种就是找加起来为$14$，例如$x$为$7$，那么可以在另一边找和$x$十进制位相同时有多少个$7$。这个方法在枚举$ve1$和遍历$x$的位数的时候都还好，但是对于在$ve2$中查找的时候就需要对$ve2$不停的排序二分查找，但是如果直接二分查找或者怎么弄常数都会非常大。
• 这个题还有一点就是使用基数排序，排序之后设$ve1$和$ve2$中的低$k$位已经排序好了，假如这个时候在找第$k+1$位会组成有多少个$4$的时候可以直接分成$ve1$中每个$k+1$位和$ve2$中的$k+1$直接加起来变成$4$或$14$，以及$ve1$中每个$k+1$位和$ve2$中的$k+1$加起来加上第$k$位的进位变成$4$或$14$。这里为什么要这么分呢，因为这样分双方都是有序的数这样就不需要二分查找，直接一次遍历就行了，减小了复杂度。
• 这题的时间卡得有点紧，难受。

---

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll maxn = 1e6+100;

struct Node {//r为已经排好序的低位
    ll l, r;
};

vector <Node> vel, ver;
vector <Node> butl[15], butr[15];

ll n, t, num[100];

void init() {
    vel.clear();
    ver.clear();
    scanf("%lld", &n);
    for(ll i=1;i<=n;i++) {
        scanf("%lld", &num[i]);
    }

    for(ll i=0;i<(1<<(n/2));i++) {
        ll sum = 0;
        ll pos = 1;
        for(ll j=1;j<=i;j<<=1) {
            if(i&(j)) {
                sum+=num[pos];
            }
            pos++;
        }
        vel.push_back({sum, 0});
    }

    for(ll i=0;i<(1<<(n - n/2));i++) {
        ll sum = 0;
        ll pos = n/2+1;
        for(ll j=1;j<=i;j<<=1) {
            if(i&(j)) {
                sum+=num[pos];
            }
            pos++;
        }
        ver.push_back({sum, 0});
    }
}

ll get_half_ans(vector <Node> &l, vector <Node> &r, ll base) {
    ll pos = 1ll * r.size() - 1;
    ll ans = 0;
    for(ll i=0;i<l.size();i++) {
        while(pos >= 0 && l[i].r + r[pos].r >= base) pos--;
        ans += pos + 1;
    }
    return ans;
}

ll get_all_ans(vector <Node> &l, vector <Node> &r, ll base) {
    ll pos = 0;
    ll ans = 0;
    for(ll i=1ll*l.size()-1;i>=0;i--) {
        while(pos < r.size() && r[pos].r + l[i].r < base) pos++;
        ans += r.size()-pos;
    }
    return ans;
}

int main() {
//    freopen("1.in.txt", "r", stdin);
    scanf("%lld", &t);
    while(t--) {
        init();

        ll base = 1;
        ll ans = 0;
        for (ll k = 0; k <= 8; k++) {
            for(ll i=0;i<vel.size();i++) {
                Node x = vel[i];
                ll w = x.l % 10;
                butl[w].push_back({x.l / 10, x.r});
            }
            for (ll i=0;i<ver.size();i++) {
                Node x = ver[i];
                ll w = x.l % 10;
                butr[w].push_back({x.l / 10, x.r});
            }

            for (ll i = 0; i <= 9; i++) {
                ll x1 = (14 - i) % 10, x2 = (13 - i) % 10;
                ans += get_half_ans(butl[i], butr[x1], base) + get_all_ans(butl[i], butr[x2], base);
            }

            ll Index = 0;
            for (ll i = 0; i <= 9; i++) 
                for (ll j = 0; j < butl[i].size(); j++) 
                    vel[Index++] = {butl[i][j].l, i * base + butl[i][j].r};
            Index = 0;
            for (ll i = 0; i <= 9; i++) 
                for (ll j = 0; j < butr[i].size(); j++) 
                    ver[Index++] = {butr[i][j].l, i * base + butr[i][j].r};

            for (ll i = 0; i <= 9; i++) butl[i].clear(), butr[i].clear();
            base *= 10;
        }
        printf("%lld\n", ans);
    }
    return 0;
}