<span>codeforces 1373F Network Coverage</span>
题意
给一个\(n\)个点的环,每个点是一个容量为\(a[i]\)的空水池,给\(n\)个水桶,第\(i\)个桶里有\(b[i]\)升水,第\(i\)个桶只能给第\(i\)和第\(i+1\)个水池加水,特殊的,第\(n\)个桶只能给第\(n\)个和第\(1\)个水池加水,问是否能将所有水池加满。
分析
二分\(b[1]\)给\(a[1]\)加了多少水,我们可以贪心的得到\(b[1]\)给\(a[2]\)加了多少水,\(b[2]\)给\(a[2]\)加了多少水....
有两种情况:
- 断流,因为\(b[1]\)给\(a[1]\)加了太多的水,导致无法加满\(a[2],a[3],\dots,a[n]\),要减少\(b[1]\)对\(a[1]\)的供水量。
- 能加满\(a[2],a[3],\dots,a[n]\),考虑增加\(b[1]\)对\(a[1]\)的供水量,因为如果仍能加满\([2,n]\)的水池,在贪心过程中因为水池容量限制造成的损耗会减少,则\(b[1]\)和\(b[n]\)对\(a[1]\)的总供水量会增加。
Code
#include<algorithm>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<sstream>
#include<cstdio>
#include<string>
#include<vector>
#include<bitset>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define rep(i,x,n) for(int i=x;i<=n;i++)
#define per(i,n,x) for(int i=n;i>=x;i--)
#define sz(a) int(a.size())
#define rson mid+1,r,p<<1|1
#define pii pair<int,int>
#define lson l,mid,p<<1
#define ll long long
#define pb push_back
#define mp make_pair
#define se second
#define fi first
using namespace std;
const double eps=1e-8;
const int mod=1e9+7;
const int N=1e6+10;
const int inf=1e9;
int t,n,a[N],b[N];
int gao(int x){
int res=b[1]-x;
rep(i,2,n){
res=b[i]-max(a[i]-res,0);
if(res<0) return -1;
}
return res;
}
int main(){
ios::sync_with_stdio(false);
//freopen("in","r",stdin);
cin>>t;
while(t--){
cin>>n;
rep(i,1,n) cin>>a[i];
rep(i,1,n) cin>>b[i];
int l=0,r=b[1],ans=0;
while(l<=r){
int mid=l+r>>1;
int ret=gao(mid);
if(ret==-1){
r=mid-1;
}else l=mid+1;
if(ret!=-1&&ret+mid>=a[1]){
ans=1;
}
}
if(ans) puts("YES");
else puts("NO");
}
return 0;
}