题目大意:
zstu的萌新们准备去自助旅行,他们租了一辆吉普车,然后选择了n个城市作为游览地点。然后他们惊喜的发现他们选择的城市刚好绕城一个环。
也就是说如果给所有城市按照0,1,2,……,n-1编号,0号城市和n-1号城市是相邻的,并且只能从i号城市去(i+1)%n号城市。
已知每个城市可以充油gas(i),从 i 到 (i+1)%n 城市耗油 cost(i)。n<=1e5;
假设这辆吉普车没有的油箱一开始是空的,并且没有上限。
没有油的话自然就不能继续旅行了,这个问题让萌新们非常困扰。作为优秀的acmer,请你帮他们找到一个出发城市,使得萌新们能游览尽可能多的城市(注意最多游览n个城市)。如果有多个可选择的出发城市,那么请把他们按照编号从小到大输出。
思路:先将每个城市的充油减去去下一个城市的耗油,为了方便我们将数组扩展为两倍。那么问题就变成了
给你n个数字,任选一个a点为起点,这个起点能达到的最远距离b,b是第一个满足sum[b]-sum[a-1]<0,那么
我们先求一个前缀和然后用单调栈就能解决问题了,时间复杂度O(n)。
ps:又忘了初始化,wa哭了。
#include<bits/stdc++.h>
using namespace std;
const int N=4e5+;
int a[N],n,ans[N],sum[N],r[N],tot,st[N];
vector<int> out;
void init()
{
out.clear();
for(int i=;i<=n;i++) ans[i]=n-;
memset(sum,,sizeof(sum));
memset(r,,sizeof(r));
tot=;
}
int main()
{
int T; scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
init();
for(int i=;i<=n;i++) scanf("%d",&a[i]);
for(int i=;i<=n;i++)
{
int g; scanf("%d",&g);
a[i]-=g;
a[i+n]=a[i];
}
for(int i=;i<=*n;i++) sum[i]+=sum[i-]+a[i];
for(int i=*n;i>=;i--)
{
while(tot> && sum[st[tot-]]>=sum[i]) tot--;
r[i]=tot== ? *n+:st[tot-];
st[tot++]=i;
}
int mx=;
for(int i=;i<n;i++) ans[i+]=min(ans[i+],r[i]-i-),mx=max(ans[i+],mx);
for(int i=;i<=n;i++) if(ans[i]==mx) out.push_back(i-);
printf("%d",out[]);
int len=out.size();
for(int i=;i<len;i++) printf(" %d",out[i]);
puts("");
}
return ;
}