Description
魔法师小Q拥有n个宝石,每个宝石的魔力依次为w_1,w_2,…,w_n。他想把这些宝石镶嵌到自己的法杖上,来提升
法杖的威力。不幸的是,小Q的法杖上宝石镶嵌栏太少了,他必须扔掉k个宝石才能将剩下的宝石镶嵌上去。法杖的
威力等于镶嵌在上面的所有宝石的魔力按位做或(OR)运算的结果,请写一个程序帮助小Q做出最佳的选择,使得法
杖的威力最大。
Input
第一行包含两个正整数n,k(2<=n<=100000,1<=k<=100,k
设最大的数为 w,若 n > k +log w,那么显然所有 1 都可以保留,否则现在 n ≤ k +log w。
考虑 DP,设 fi,j 表示考虑前 i 个数,保留的数的 or 是 j 时,最多能删除多少个数。
时间复杂度 O(n + (k + log w)w)。
题解如上。
差不多了,只不过直接dp不大好写,f[i][j]表示用最少的数能到达j状态,转移就很好写了。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
const int inf=0x3f3f3f3f;
int n,m,mx,k;
int a[N];
int f[200][200000],bin[N];
int main()
{
scanf("%d%d",&n,&k);
fo(i,1,n)scanf("%d",&a[i]),mx=max(mx,a[i]);
if (n>k+log(mx)/log(2))
{
int ans=0;
fo(i,1,n)ans|=a[i];
printf("%d\n",ans);
return 0;
}
memset(f,inf,sizeof(f));
f[0][0]=0;
bin[0]=1;
fo(i,1,17)bin[i]=bin[i-1]*2;
fo(i,0,n-1)
fo(j,0,bin[17]-1)if (f[i][j]<inf)
{
f[i+1][j]=min(f[i+1][j],f[i][j]);
f[i+1][j|a[i+1]]=min(f[i+1][j|a[i+1]],f[i][j]+1);
}
int ans=0;
fd(i,bin[17]-1,0)
if (f[n][i]<=n-k)
{
ans=i;
break;
}
printf("%d\n",ans);
}