CodeForces - 1316E Team Building(状压dp)
題目鏈接:點擊查看
題目大意:給出 n 個人,現在需要 p 個球員和 k 個啦啦隊員,n 個人當啦啦隊員以及當第 i 個位置的球員所做的貢獻都會給出,現在問如何分配可以使得總貢獻和最大
題目分析:因為 p 最大只有 7 ,不難想到狀態壓縮,而壓縮的每個狀態可以表示球員每個位置的占有情況,對于啦啦隊員,我們可以貪心排序,從最大的進行遍歷,dp[ i ][ j ] 代表到了第 i 個人為止,狀態為 j 時的總貢獻最大值,轉移方程分為三個部分,第一個部分是這個人不會被選,顯然dp[ i ][ j ] = dp[ i - 1 ][ j ],第二個部分是當球員,根據二進制的信息可以輕松轉移最大值,最難的部分我感覺是處理啦啦隊員的信息了,首先我們預處理出一個cnt數組,用來表示每個二進制狀態中有多少個 1 ,也就是代表了每個狀態中已經選出了多少個球員,對于某個位置 i 和某個狀態 j 來說,此時之前一共有 i - 1 個人,cnt[ j ]個球員,那么已經選了 i - 1 - cnt[ j ] 個啦啦隊員了,如果 i? - 1 - cnt[ j ] + 1 <=?k 的話,也就是 i - cnt[ j ] <= k 的話,那么這個位置就還缺少啦啦隊員,因為經過了之前的排序操作,越早選必定是越優的,注意,此時的 cnt[ j ] 必須滿足嚴格小于 i 才行,不然dp的意義來說,如果 cnt[ j ] >= i 的話,那就沒有多余的人供啦啦隊選擇了
代碼:
#include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e5+100;struct Node {LL a,s[7];bool operator<(const Node& t)const{return a>t.a;} }a[N];LL cnt[(1<<7)+10],dp[2][(1<<7)+10];void init() {for(int i=0;i<(1<<7);i++)cnt[i]=cnt[i>>1]+(i&1); }int main() { #ifndef ONLINE_JUDGE // freopen("input.txt","r",stdin); // freopen("output.txt","w",stdout); #endif // ios::sync_with_stdio(false);init();int n,p,k;scanf("%d%d%d",&n,&p,&k);for(int i=1;i<=n;i++)scanf("%lld",&a[i].a);for(int i=1;i<=n;i++)for(int j=0;j<p;j++)scanf("%lld",&a[i].s[j]);sort(a+1,a+1+n);for(int i=1;i<=n;i++){int now=i&1;for(int j=0;j<(1<<p);j++){dp[now][j]=dp[now^1][j];//什么都不干for(int t=0;t<p;t++)//選擇當某個位置的球員 if((j>>t)&1)dp[now][j]=max(dp[now][j],dp[now^1][j^(1<<t)]+a[i].s[t]);if(cnt[j]<i&&i-cnt[j]<=k)//當拉拉隊員 dp[now][j]=max(dp[now][j],dp[now^1][j]+a[i].a); // cout<<i<<' '<<j<<' '<<dp[now][j]<<endl;}}printf("%lld\n",dp[n&1][(1<<p)-1]);return 0; }?
總結
以上是生活随笔為你收集整理的CodeForces - 1316E Team Building(状压dp)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: CodeForces - 1316D N
- 下一篇: 中石油训练赛 - Match Match