题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6682
题意:现在有n个数,你可以任意组合,问在所有的组合中,对每个组合求和,那么总共含有多少个数字4。
解题心得:
- 看到 n n n最大为 40 40 40,第一个反应肯定就是折半搜索,首先将所有的 n n n个数均分成两半,并且分别将每一半的所有组合求和暴力枚举出来,设存放在 v e 1 ve1 ve1和 v e 2 ve2 ve2中。接下来就可以枚举 v e 1 ve1 ve1中的每一个数设当前数为 x x x,枚举 x x x中的每个位数,然后可以直接在另一半中找有多少个可以和 x x x加起来变成 4 4 4,如 x x x等于 2 2 2时可以在另一边找和 x x x十进制位相同时有多少个 2 2 2,第二种就是找加起来为 14 14 14,例如 x x x为 7 7 7,那么可以在另一边找和 x x x十进制位相同时有多少个 7 7 7。这个方法在枚举 v e 1 ve1 ve1和遍历 x x x的位数的时候都还好,但是对于在 v e 2 ve2 ve2中查找的时候就需要对 v e 2 ve2 ve2不停的排序二分查找,但是如果直接二分查找或者怎么弄常数都会非常大。
- 这个题还有一点就是使用基数排序,排序之后设 v e 1 ve1 ve1和 v e 2 ve2 ve2中的低 k k k位已经排序好了,假如这个时候在找第 k + 1 k+1 k+1位会组成有多少个 4 4 4的时候可以直接分成 v e 1 ve1 ve1中每个 k + 1 k+1 k+1位和 v e 2 ve2 ve2中的 k + 1 k+1 k+1直接加起来变成 4 4 4或 14 14 14,以及 v e 1 ve1 ve1中每个 k + 1 k+1 k+1位和 v e 2 ve2 ve2中的 k + 1 k+1 k+1加起来加上第 k k k位的进位变成 4 4 4或 14 14 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;
}