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