题目
n(n<=500)种球,第i种有ai(0<=ai<=1e12)个球,
m(m<=5e5)个盒子,第j个能放bj(0<=bj<=1e12)个球
特别地,第j个盒子最多能放i*j个第i种球
求m个盒子能放的最多的球的总数
思路来源
官方题解
题解
显然是一个最大流模型,超级源点s到超级汇点的流量t,
由于最小割=最大流,可以考虑最后这个图,割完之后长什么样
比如左侧1、3记为集合P含于S,右侧点2记为集合Q含于S,
那么,记左侧集合非P含于T,右侧集合非Q含于T
那么,最小割的边集的构成,由三部分组成:
1. 超级源点s与集合非P之间的边,即左侧属于t的点,断开与s的边
2. 集合Q与超级汇点t之间的边,即右侧属于s的点,断开与t的边
3. 左侧集合P与右侧集合非Q之间的边,左侧属于s的点,右侧属于t的点,断开左右点之间的边
由于边是有向的,
所以无需断开左侧属于t的点和右侧属于s的点之间的边,
因为从上游流量就已经切断了
然后就是对官方题解的一些补充说明吧,
最小割的代价由三部分组成,
形如
所以枚举,也就是左侧属于S集合的i之和,
这样可以通过dp,O(n^3)求得,
也就是属于S集合i之和固定时,不属于S集合的Ai之和的最小值
而后面两坨,k固定时,答案之和j有关,
可以任意划分,将一部分划给S集合,另一部分划给T集合,
并且划给S集合的每个点贡献是j*k,划给T集合的每个点贡献是B[j],
使得这两部分之和最小,那么考虑某一个点,自然是哪个小划给哪边,
所以每个点贡献是min(j*k,B[j])
由j*k>B[j],解得k>=B[j]/j,所以枚举k的时候,每个点从S换到T的操作只会发生一次
记录一下这个翻转的时机,即可一边枚举k一边实现对贡献的统计,
这部分复杂度O(n^2+m)
总复杂度O(n^3+n^2+m)
代码
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<ll,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define scll(a) scanf("%lld",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=505,M=5e5+10,S=N*(N+1)/2;
const ll INF=0x3f3f3f3f3f3f3f3fll;
int n,m;
ll a[N],b[M],dp[N][S],ans,sum,sum2;
vector<int>flip[S];
void upd(ll &x,ll y){x=min(x,y);
}
int main(){sci(n),sci(m);rep(i,1,n)scll(a[i]);rep(i,1,m)scll(b[i]);memset(dp,INF,sizeof dp);dp[0][0]=0;rep(i,0,n-1){int up=i*(i+1)/2,v=i+1;rep(j,0,up){upd(dp[i+1][j+v],dp[i][j]);upd(dp[i+1][j],dp[i][j]+a[i+1]);}}int lim=n*(n+1)/2;rep(j,1,m){//先认为都是j*k,再翻到b[j]ll v=b[j]/j;if(v<=lim)flip[v].pb(j);sum+=j;}ans=8e18;rep(j,0,lim){ans=min(ans,dp[n][j]+1ll*sum*j+sum2);for(auto &v:flip[j]){sum-=v;sum2+=b[v];}}printf("%lld\n",ans);return 0;
}