題目大意：給出 n 個武器，每個武器可以升級最多 k[ i ] 次，每次升級需要花費 c[ i ][ j ] 個單位的金幣，性能可以提升 w[ i ][ j ] 個單位，現在給出 m 個金幣，問如何分配可以使得總性能最大

題目分析：n 只有 20，k 只有 4，直接搜索的話時間復雜度為 5^20，顯然是不可行的，之前做過一個模型一樣的題目：AcWing - 171，所以不難想到可以寫雙向dfs+二分，直接實現就好了

有一個坑點就是，在儲存了數組 c 和 w 后，因為需要求前綴和，4 * 1e9 會爆掉 int，所以全部涉及到數值的地方都要開 long long

代碼：
?

//#pragma GCC optimize(2) //#pragma GCC optimize("Ofast","inline","-ffast-math") //#pragma GCC target("avx,sse2,sse3,sse4,mmx") #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> #include<cassert> #include<bitset> #include<list> #include<unordered_map> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=1e6+100;int n,m,limit,num[N];LL cost[N][5],val[N][5];LL ans;vector<pair<LL,LL>>node;//(cost,val)void dfs1(int step,LL c,LL v)//搜1~limit {if(step>limit){node.emplace_back(c,v);return;}for(int i=0;i<=num[step];i++)//0~num[step],0的意思是不選if(c+cost[step][i]<=m)dfs1(step+1,c+cost[step][i],v+val[step][i]); }LL get_pos(int v)//小于等于v的最大值 {int l=0,r=node.size()-1;LL ans=0;while(l<=r){int mid=l+r>>1;if(node[mid].first<=v){l=mid+1;ans=node[mid].second;}elser=mid-1;}return ans; }void dfs2(int step,LL c,LL v)//搜limit+1~n {if(step>n){ans=max(ans,v+get_pos(m-c));return;}for(int i=0;i<=num[step];i++)//0~num[step],0的意思是不選if(c+cost[step][i]<=m)dfs2(step+1,c+cost[step][i],v+val[step][i]); }int main() { #ifndef ONLINE_JUDGE // freopen("data.in.txt","r",stdin); // freopen("data.ans.txt","w",stdout); #endif // ios::sync_with_stdio(false);int w;cin>>w;int kase=0;while(w--){node.clear();ans=0;scanf("%d%d",&n,&m);limit=n/2;for(int i=1;i<=n;i++){scanf("%d",num+i);for(int j=1;j<=num[i];j++){scanf("%lld%lld",val[i]+j,cost[i]+j);cost[i][j]+=cost[i][j-1];val[i][j]+=val[i][j-1];}}dfs1(1,0,0);sort(node.begin(),node.end());node.erase(unique(node.begin(),node.end()),node.end());for(int i=1;i<node.size();i++)node[i].second=max(node[i].second,node[i-1].second);dfs2(limit+1,0,0);printf("Case #%d: %lld\n",++kase,ans);}return 0; }

?

總結

以上是生活随笔為你收集整理的中石油训练赛 - Get Strong(dfs双向搜索+二分)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。