Description
有 个城市围成一圈,第 个城市距离第 个城市 ,第 个城市距离第 个城市 ,第 个城市的加油站一次可以加行驶 距离的油,初始油箱里没有油,问从哪些城市出发(任选方向)可以完整的依次走过这 个城市
Input
第一行一整数 表示城市个数,之后输入 个整数 表示每个城市加油站可以加的油量,最后输入 个整数 表示城市之间的距离,保证两序列和相同
Output
输出满足条件的城市数量及其编号
Sample Input
4
1 7 2 3
8 1 1 3
Sample Output
2
2 4
Solution
先单方向 考虑,另一方向类似,以 表示从第 个城市出发经过这 个城市期间的油量最小值,显然若 说明从第 个城市出发可以满足条件,由于加油总量和路程总量相同,假设从 城市出发,最终在 城市油量必然为 ,那么 的取值即为从 出发在 城市时油量与 之间的最小值,而 的取值为从 出发在 城市时油量的最小值,这相当与给之前 个值加上 后再取最小值,故有 ,任意固定起点得到初值后即可 完成转移
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
ll dp[maxn];
int n,a[maxn],b[maxn],vis[maxn];
vector<int>ans;
void Solve()
{
ll temp=0,mn=0;
for(int i=1;i<=n;i++)
{
temp+=a[i]-b[i];
mn=min(mn,temp);
}
dp[1]=dp[n+1]=mn;
for(int i=n;i>1;i--)dp[i]=dp[i+1]+b[i]-a[i];
for(int i=1;i<=n;i++)
if(dp[i]>=0)vis[i]=1;
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++)scanf("%d",&b[i]);
memset(vis,0,sizeof(vis));
ll temp=0,mn=0;
for(int i=1;i<=n;i++)
{
temp+=a[i]-b[i];
mn=min(mn,temp);
}
dp[1]=dp[n+1]=mn;
for(int i=n;i>1;i--)dp[i]=dp[i+1]+a[i]-b[i];
for(int i=1;i<=n;i++)
if(dp[i]>=0)vis[i]=1;
temp=0,mn=0;
b[0]=b[n];
for(int i=n;i>=1;i--)
{
temp+=(a[i]-b[i-1]);
mn=min(mn,temp);
}
dp[0]=dp[n]=mn;
for(int i=1;i<n;i++)dp[i]=dp[i-1]+a[i]-b[i-1];
for(int i=1;i<=n;i++)
if(dp[i]>=0)vis[i]=1;
ans.clear();
for(int i=1;i<=n;i++)
if(vis[i])ans.push_back(i);
printf("%d\n",ans.size());
for(int i=0;i<ans.size();i++)printf("%d%c",ans[i],i==ans.size()-1?'\n':' ');
}
return 0;
}