枚举在某些题目中有着非常重要的作用。
其中最方便的就是利用位运算来进行枚举,枚举2的n次方
例题:
这是一道典型的折半枚举例题,通过先枚举其中一半的数据,在通过数组储存起来。这样就可以进行排序后的二分搜索了。
空间换时间
int n2 = n/2;
for (int i = 0;i < 1 << n2;i ++){
ll sw = 0,sv = 0;
for (int j = 0;j < n2;j ++){
if (i >> j & 1){
sw += w[j];
sv += v[j];
}
}
ps[i] = make_pair(sw,sv);
}
这是其中枚举的重要一步,假设我们枚举两个物品的选择情况,那么总共是2的2次方个。分别是 00 ,01,10,11这样我们就可以通过 for i 0 -> 1<<2 就把所有的情况都枚举出来了,但这样还不够,我们还要访问每个物品的情况,这样的话,我们就可以通过 for j 0 -> 2 来访问每个物品的情况。
具体的代码如下
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAX_N = (int) 40;
const ll INF = (ll)0x3f3f3f3f;
// 输入
int n;
ll w[MAX_N],v[MAX_N];
ll W;
pair<ll,ll> ps[1 << (MAX_N/2)]; // 重量价值
void solve(){
//枚举前半部分
int n2 = n/2;
for (int i = 0;i < 1 << n2;i ++){
ll sw = 0,sv = 0;
for (int j = 0;j < n2;j ++){
if (i >> j & 1){
sw += w[j];
sv += v[j];
}
}
ps[i] = make_pair(sw,sv);
}
//除去多余元素
sort(ps,ps + (1<<n2));
int m = 1;
for (int i = 1;i < 1<<n2;i ++){
if (ps[m-1].second < ps[i].second){
ps[m++] = ps[i];
}
}
// 枚举后半部分并求解
ll res = 0;
for (int i = 0;i < 1<<(n-n2);i ++){
ll sw = 0,sv = 0;
for (int j = 0;j < n-n2;j ++){
if (i >> j & 1){
sw += w[n2+j];
sv += v[n2+j];
}
}
if (sw <= W){
ll tv = (lower_bound(ps,ps+m,make_pair(W - sw,INF))-1)->second;
res = max(res,sv + tv);
}
}
printf("%lld\n",res);
}
int main()
{
cin >> n;
for (int i = 0;i < n;i ++)cin >> w[i];
for (int i = 0;i < n;i ++)cin >> v[i];
cin >> W;
solve();
}