NOIP模拟——偷书（读书人的事，那能叫偷吗？）

题目

在L的书架上，有N本精彩绝伦的书籍，每本书价值不菲。

M是一个书籍爱好者，他对L的书籍早就垂涎三尺。最后他忍受不了诱惑，觉得去偷L的书，为了迅速完成这件事，同时他不希望L很快发现书籍少了，他决定偷书时，对于任意连续的k本书，他最多选B本，最少选A本。现在他想知道怎么选出来的书本最后使得偷的书籍的价值和，与剩下的书籍价值和，差值最大。

输入

第一行四个整数 n,k,a,b

一行 N 个整数表示每本书的价值

输出

一个整数表示答案

样例输入

2 1 0 1
2 -2

样例输出

4

提示

得到第一本书得到的价值和是 2

剩余的价值和是-2

差值为 4

对于 20%：n<=10

对于另外 20%：a=0,b=k

对于 100%：n<=1000,0<=a<=b<= k<=10 ，所有书籍的价值的绝对值<=10^9

看到k小于等于10就应该想到状压dp的

dp[i][j]表示当前在第i个，状态为j的最优解，因为每次向后递推一个，我们只需要维护最后一个是否选了，并且判断现在这个选不选

结果被数组大小给卡了，真是那啥了那啥

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define ll long long
 4 int v[1055],n,k,b,a,num[1024];
 5 ll dp[1055][1024],sum;
 6 inline void getn(){
 7     for(int i=0;i<1024;i++)
 8     {
 9         int t=i,tot=0;
10         while(t)
11         {
12             if(t&1)
13             {
14                 tot++;
15             }
16             t/=2;
17         }
18         num[i]=tot;
19     }
20 }
21 inline bool judge(int k)
22 {
23     return !(num[k]<a||num[k]>b);
24 }
25 int main(){
26     scanf("%d%d%d%d",&n,&k,&a,&b);
27     getn();
28     for(int i=1;i<=n;i++)
29     {
30         scanf("%d",&v[i]);sum+=v[i];
31     }
32     for(int i=0;i<(1<<k);i++)
33     {
34         for(int j=0;j<k;j++)
35         {
36             if(i&(1<<j)) dp[k][i]+=v[k-j];
37         }
38     }
39     for(int i=k+1;i<=n;i++)
40     {
41         for(int j=0;j<(1<<k);j++)
42         {
43             if(!judge(j)) continue;
44             int st=j;
45             st>>=1;
46             if(j&1)
47             {
48                 int sta=st+(1<<k-1);
49                 dp[i][j]=dp[i-1][sta]+v[i];
50                 if(judge(st)) dp[i][j]=max(dp[i][j],dp[i-1][st]+v[i]);
51             }
52             else
53             {
54                 int sta=st+(1<<k-1);
55                 dp[i][j]=dp[i-1][st];
56                 if(judge(sta)) dp[i][j]=max(dp[i][j],dp[i-1][sta]);
57             }
58         }
59     }
60     ll ans=-1000000000000;
61     for(int i=0;i<(1<<k);i++)
62     {
63         if(judge(i))
64         ans=max(ans,dp[n][i]);
65     }
66     cout<<ans-sum+ans<<endl;
67     return 0;
68 }