神贪心……写了一个晚上加一个早上。
先考虑只有一个宿管的情况。
- 首先,如果这个宿舍人多了,多余的人就跑到下一个宿舍。(如果这是最后一个宿舍的话,多的就躺床底下)
- 如果这个宿舍人少了,但是能从别的宿舍调过来人,那就调人。
- 如果这个宿舍人少了,从别的宿舍也调不过来足够的人,那就全跑到下一个宿舍去。
- 让宿管查宿。
关于调人,我们可以每次总是从当前能调过来人的最远宿舍调人,调成负的也无所谓。想一想,为什么。
还有一个神奇的事情:不同的人的路径不交。比如说,有 \(i \leq a \leq b \leq j\),要是 \(i\) 跑到 \(b\),\(j\) 跑到 \(a\),就不如 \(i\) 去 \(a\),\(j\) 去 \(b\)。
因此,有这样一个宿舍:这个宿舍左边的人都去对付第一个宿管,右边的人都去对付第二个宿管,这个宿舍的人有去对付第一个的也有去对付第二个的。
或者说,我们枚举有多少个人对付第一个宿管。每次枚举都 \(\mathrm{O}(n)\) 的求一下,总复杂度是 \(\mathrm{O}(n^2b)\),这样不好,考虑加速枚举。
可以发现,分去对付第一个宿管的人越多,宿管发现的赖宿舍数越少。
记 \(f(m)\) 是给第一个宿管分去 \(m\) 人然后发现的赖宿舍数,显然 \(f(m)\) 是非严格单调减的。记 \(g(m)\) 是给第一个宿管分去 \(m\) 人(也就是给第二个宿管分 \(nb-m\) 人)然后第二个宿管发现的赖宿舍数,显然 \(g(m)\) 是非严格单调增的。
欲最小化 \(\max(f(m),g(m))\),则记 \(z(m)=g(m)-f(m)\),则 \(z(m)\) 非严格单调增。二分找出那个合适的 \(m\) 即可。
需要注意的一点是,如果 \(z(m)=0\) 那当然好,那样答案就是 \(ans(m)\)。可是我们在二分的时候写的判定是 \(z(m) \geq 0\) 这种,有可能找出来的是 \(z(m) > 0\) 的最小 \(m\),因此还要看看 \(ans(m-1)\),看看到底哪个更小。
#include
#include
#include
using namespace std;
int aa[100005], n, d, b, qq[100005], cur=1, a[100005], mid;
int f(int hmn, int far){
int r=1, sum=0, re=0;
for(int i=1; i<=hmn; i++){
for(; r<=far && r<=i+d*i; r++)
sum += a[r];
if(a[i]>=b){
int tmp=a[i]-b;
a[i] -= tmp;
a[i+1] += tmp;
sum -= a[i];
}
else{
if(sum>=b){
a[r-1] -= b - a[i];
a[i] = b;
sum -= a[i];
}
else{
a[i+1] += a[i];
a[i] = 0;
re++;
}
}
}
return re;
}
int z(int x, int &tmp1, int &tmp2){
while(qq[cur-1]>=x && cur>1) cur--;
while(qq[cur]=cur; i--)
a[n-i+1] = aa[i];
a[n-cur+1] = qq[cur] - x;
tmp2=f(n/2, n-cur+1);
return tmp2-tmp1;
}
int main(){
cin>>n>>d>>b;
for(int i=1; i<=n; i++){
scanf("%d", &aa[i]);
qq[i] = qq[i-1] + aa[i];
}
int l=0, r=n*b, re, tmp1, tmp2;
while(l<=r){
mid = (l + r) >> 1;
if(z(mid, tmp1, tmp2)>=0) re = mid, r = mid - 1;
else l = mid + 1;
}
z(re, tmp1, tmp2);
int ans=max(tmp1,tmp2);
z(re-1, tmp1, tmp2);
ans = min(ans, max(tmp1, tmp2));
cout<
upd:从神犇代码学的
#include
#include
using namespace std;
typedef long long ll;
int n, cnt1, cnt2, b, d;
ll a[100005];
ll sum(ll l, ll r){
if(l<1) l = 1;
if(r>n) r = n;
return a[r] - a[l-1];
}
int main(){
cin>>n>>d>>b;
for(int i=1; i<=n; i++){
scanf("%I64d", &a[i]);
a[i] += a[i-1];
}
for(int i=1; i<=n/2; i++){
if(sum(1, i+(ll)i*d)>=(ll)(cnt1+1)*b) cnt1++;
if(sum(n-i-(ll)i*d+1, n)>=(ll)(cnt2+1)*b) cnt2++;
}
cout<