1. 程式人生 > >【區間DP && 按位貪心】CodeForces

【區間DP && 按位貪心】CodeForces

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; }