【區間DP && 按位貪心】CodeForces
阿新 • • 發佈:2018-12-23
Step1 Problem:
給你 n 本書以及書的權值 a[i],讓你放到 k 個書架上(放到書架上的書必須是連續的),每個書架的值為書的權值和,輸出最大的書架的值異或和。
資料範圍:
1<=k, n <= 50, 0 < a[i] < 2^50.
Step2 Ideas:
求最大的書架的值異或和,按位從高到低列舉答案,因為高位滿足答案 肯定比 高位不滿足所有低位滿足的 結果都大。例:2^3 > 2^0 + 2^1+2^2
狀態 dp[i][k]:前 i 本書放到 k 個書架的異或和是否滿足答案
Step3 Code:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 60;
int n, K, dp[N][N];
ll a[N];
bool solve(ll ans)
{
memset(dp, 0, sizeof(dp));
dp[0][0] = 1;
for(int k = 1; k <= K; k++)
{
for(int i = 0; i <= n; i++)
{
for(int j = i+1; j <= n; j++)
{
ll t = a[j]-a[i];//書 i 到 j 的權值和
if(dp[i][k-1] && ((t&ans) == ans))//這些書新開一個書架能滿足條件
{
dp[j][k] = 1;
}
}
}
}
return dp[n][K];
}
int main()
{
while(~scanf("%d %d", &n, &K))
{
a[0] = 0;
for(int i = 1 ; i <= n; i++)
{
scanf("%lld", a+i);
a[i] += a[i-1];
}
ll ans = 0;
for(int i = 60; i >= 0; i--)//從高位列舉到低位
{
if(solve(ans|(1LL<<i)))//如果加上該位能滿足條件,我們當然會要
{
ans |= (1LL<<i);
}
}
printf("%lld\n", ans);
}
return 0;
}