D. River Locks(思维+简单dp)

最新推荐文章于 2022-07-20 13:56:40 发布

要用bug来打败bug

最新推荐文章于 2022-07-20 13:56:40 发布

阅读量451

点赞数 2

分类专栏： acm 文章标签：动态规划算法 c++

本文链接：https://blog.csdn.net/g3125976202/article/details/125383129

版权

acm 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

D. River Locks

题目大意：

$n$ 个水槽体积是 $v$ , $q$ 个查询，每次给定注水时间 $t i$ ，问最小需要同时开几个水管，在 $t i$ 前注满所有水槽。（一个水槽注满会向下一个转移时间是0）

思路：

时间考虑到 $t i$ （没什么好说的）
设选了 $x$ 个水槽，那这 $x$ 个水槽最好选在 $1 - x$ 。（水满转移不需要时间，转移只能向后，优先满足左边）
那不就是总体积除于 $t i$ 上取整？？
是但不完全是！要满足选的前 $x$ 个水槽在 $t i$ 时刻都已充满，否则就是 $- 1$ 。（为什么不扩大 $x$ 这个以后说）
$d p [i]$ 记录的是充满前 $i$ 个水槽的最段时间。
$d p [1] = v [1]$
在第 $i$ 个水槽时，直接在 $d p [i - 1]$ 的时间基础上二分即可。

void work(){
	dp[1]=v[1],sum[1]=v[1];
	for(int i=2;i<=n;i++){
		sum[i]=sum[i-1]+v[i];  // 前缀和
		ll l=dp[i-1],r=1e9;
		while(l<r){
			ll mid=l+r>>1;
			if(mid*i-sum[i-1]>=v[i]) r=mid;  // 前i-1都已满足，余下的水一定可以转移
			else l=mid+1;
		}
		dp[i]=l;
	}
}

不难发现，整个 $d p$ 数组是非严格上升的，故当最小的 $x$ 不满足时，答案是 $- 1$ .

Code

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N=2e5+10;
ll n,dp[N],v[N],sum[N];

void work(){
	dp[1]=v[1],sum[1]=v[1];
	for(int i=2;i<=n;i++){
		sum[i]=sum[i-1]+v[i];
		ll l=dp[i-1],r=1e9;
		while(l<r){
			ll mid=l+r>>1;
			if(mid*i-sum[i-1]>=v[i]) r=mid;
			else l=mid+1;
		}
		dp[i]=l;
	}
}

int main(){
guo312;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>v[i];
	}
	work();
	int q; cin>>q;
	while(q--){
		ll ti; cin>>ti;
		ll ans=sum[n]/ti;
		if(sum[n]%ti) ans++;
		if(dp[ans]<=ti&&ans<=n) cout<<ans<<endl;
		else cout<<"-1"<<endl;
	}
	return 0;
}

代码 $r e$ 了？？？
注意在判断输出时是且运算符，要先判断答案是否小于等于 $n$ 。

AC代码

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N=2e5+10;
ll n,dp[N],v[N],sum[N];

void work(){
	dp[1]=v[1],sum[1]=v[1];
	for(int i=2;i<=n;i++){
		sum[i]=sum[i-1]+v[i];
		ll l=dp[i-1],r=1e9;
		while(l<r){
			ll mid=l+r>>1;
			if(mid*i-sum[i-1]>=v[i]) r=mid;
			else l=mid+1;
		}
		dp[i]=l;
	}
}

int main(){
guo312;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>v[i];
	}
	work();
	int q; cin>>q;
	while(q--){
		ll ti; cin>>ti;
		ll ans=sum[n]/ti;
		if(sum[n]%ti) ans++;
		if(ans<=n&&dp[ans]<=ti) cout<<ans<<endl;
		else cout<<"-1"<<endl;
	}
	return 0;
}