【CF981D】Bookshelves（贪心，动态规划）

【CF981D】Bookshelves（贪心，动态规划）

题面

洛谷
Codeforces
给定一个长度为\(n\)的数列，把他们划分成\(k\)段，使得每段的和的结构按位与起来最大。

题解

从高位往低位贪心，然后暴力\(dp\)就行了吧。。。

#include<iostream>
#include<cstdio>
using namespace std;
#define MAX 55
#define ll long long
inline ll read()
{
    ll x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
int n,k;ll a[MAX],s[MAX],ans;
bool f[MAX][MAX];
bool dp(ll lim)
{
    for(int i=0;i<=n;++i)
        for(int j=0;j<=k;++j)f[i][j]=false;
    f[0][0]=true;
    for(int i=1;i<=n;++i)
        for(int j=0;j<i;++j)
            for(int k=0;k<i;++k)
                if(((s[i]-s[k])&lim)==lim)
                    f[i][j+1]|=f[k][j];
    return f[n][k];
}
int main()
{
    n=read();k=read();
    for(int i=1;i<=n;++i)a[i]=read(),s[i]=s[i-1]+a[i];
    for(int i=60;~i;--i)if(dp(ans|(1ll<<i)))ans|=1ll<<i;
    cout<<ans<<endl;
    return 0;
}