题意:
给定一个数列1~n,他们头尾相接,每个数上都有一个整数ai。
可以选择1~n中的任意一个点k,使得他沿着环一圈相加不会出现负数。
求有几个点。
解析:
解法一:暴力
我们可以循环枚举每一个点,进行模拟,O(n^2)
#include <bits/stdc++.h>
using namespace std;
int a[300005],n,res;
int main()
{
cin >> n;
for (int i=1; i<=n; i++)
{
cin >> a[i];
}
for (int i=1; i<=n; i++)
{
bool ok=0;
long long ans=0;
for (int j=i; j<=n+i-1; j++)
{
if (j>n)
{
ans+=a[j-n];
}
else
{
ans+=a[j];
}
if (ans<0)
{
ok=1;
}
}
if (!ok)
{
res++;
//cout << i<<endl;
}
}
cout << res;
}
解法二:双端队列
本题的核心在于如何选择起始位置 𝑘,使得玩家在顺时针走过 𝑛−1 步后,生命值始终不为负。玩家的初始生命值为 0,每踩上一个格子就会将生命值加上该格子上的数。
我们可以通过前缀和数组来简化生命值的计算。具体步骤如下:
-
首先计算出前缀和数组 prefix_sum,其中 prefix_sum[i] 表示从位置 1 到位置 𝑖 的格子上数值的总和。
- 为了处理循环性质,我们可以将数组扩展一倍,计算 prefix_sum 到 2𝑛。⭐
- 此步骤实现代码:
for (int i = 1; i <= n; ++i) { prefix_sum[i] = prefix_sum[i - 1] + a[i - 1]; } for (int i = n + 1; i <= 2 * n; ++i) { prefix_sum[i] = prefix_sum[i - 1] + a[i - n - 1]; }
- 使用双端队列来维护可能的起始位置 𝑘。
- 遍历每一个可能的结束位置 𝑖,并检查从 𝑘 到 𝑖 的生命值是否会变为负。
-
有效性判断:
- 对于每一个结束位置 𝑖,我们需要判断队列中的最小前缀和是否满足条件,即prefix_sum[dq.front()] - prefix_sum[i - n] >= 0。通过维护一个有效的队列,确保我们只考虑那些能让生命值保持非负的起始位置。
解法三:
经过观察可以发现,只有经过负数才会退出,所以说我们可以从每个负数向前枚举,加起来的和小于0,就标记。
最后统计没被标记的数的个数。O(n*b)b为负数个数
#include <bits/stdc++.h>
using namespace std;
long long n,i,a[300010],sum,ans,b[300010],t,f[300010],j,k;
int main()
{
cin>>n;
for(i=1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
if(a[i]<0)
{
b[++t]=i;
}
}
for(i=1;i<=t;i++)
{
if(f[b[i]]==0)
{
k=a[b[i]];
f[b[i]]=1;
for(j=b[i]-1;f[j]==0;j--)
{
if(j==0)
{
j=n;
}
if(k+a[j]<0)
{
f[j]=1,k+=a[j];
}
else
{
break;
}
}
}
}
for(i=1;i<=n;i++)
{
if(f[i]==0)
{
ans++;
}
}
cout<<ans;
}
优化:
我们可以发现假如说对于1,-3,-4,2,1这样的数据,我们从-3开始从前遍历,我们将-3,1,1标记,接着遍历-4,往前遍历,接着标记-4,-3,1,1,2,可我们发现-3,1,1被标记了两次,浪费了时间,所以我们可以用并查集优化,fa[i]初始都指向自己,如果前面的数加上之前的和比0小,将fa[i]指向下一个数,这样每次遍历就只用遍历当前数的祖先就可以了。加上路径压缩,按秩合并,O(a(n)n),a为反阿克曼函数。