Educational Codeforces Round 162 (Rated for Div. 2) D. Slimes

链接：Problem - D - Codeforces

思路：先求出前缀和和后缀和 然后从1~n以每个点 i 为中点 前后进行二分 往右查找出当前位置右边到第o个位置（o是下标）的和大于 a[i] （当前的数）（如果找不到则为0） 往左查找出左边第p个位置（p是下标）到当前位置的和大于 a[i] （当前的数） （如果找不到则为0）也就表示左边 i-p个数合并就可以吞并自己 右边o-i个数合并就可以吞并自己 不过 如果左边的o-i个数都相同 例如：5 2 2 2 2 2 这样虽然每个数左右两边最差有两个数可以大于自己 但是他们的大小都相同 则不能进行合并 所以 构建两个数组 lc记录当前位置连续数的最左边的一个 rc记录当前位置连续数的最右边的一个 左边第一个不同的数 和 p 取一个最小值 如果 到了第一个数都相同 则为0 右边第一个不同的数 和 o 取一个最小值 如果 到了第n个数都相同 则为0

#include<bits/stdc++.h>
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
using namespace std;using ll = long long;using PLL = pair<ll,ll>;
const ll MAX = 1e18;const ll MIN = -1e18;const ll INF=0x3f3f3f3f;
const ll Q = 3e5+9;const ll MOD = 1e9 + 7;
ll a[Q],lcnt[Q],rcnt[Q],lc[Q],rc[Q];
ll low(ll l,ll r,ll now,ll it){
  ll ans=0;
  while(l<=r){
    ll mid=(l+r)>>1;
    if(lcnt[mid]-lcnt[it]>now){
      r=mid-1;
      ans=mid;
    }else l=mid+1;
  }
  return ans;
}
ll upp(ll l,ll r,ll now,ll it){
  ll ans=0;
  while(l<=r){
    ll mid=(l+r)>>1;
    if(rcnt[mid]-rcnt[it]>now){
      ans=mid;
      l=mid+1;
    }else r=mid-1;
  }
  return ans;
}
void solve(){
   ll n;cin>>n;
   for (ll i = 1; i <= n; i++)
    {
      cin>>a[i];
    }lc[1]=0;
    for (ll i = 2; i <= n; i++)
    {
      if(a[i]==a[i-1]) lc[i]=lc[i-1];
      else lc[i]=i;
    }rc[n]=n+1;
    for (ll i = n-1; i >= 1; i--)
    {
      if(a[i]==a[i+1]) rc[i]=rc[i+1];
      else rc[i]=i;
    }
   for (ll i = 1; i <= n; i++)
    lcnt[i]=lcnt[i-1]+a[i];
   for (ll i = n; i >= 1; i--)
    rcnt[i]=rcnt[i+1]+a[i];
   for (ll i = 1; i <= n; i++)
   {
      ll now=a[i];
      ll o=low(i+1,n,now,i);
      ll p=upp(1,i-1,now,i);
      if(o!=0 and rc[i+1]>=o and o-i>1) o=rc[i+1]+1;
      if(p!=0 and lc[i-1]<=p and i-p>1) p=lc[i-1]-1;
      if(o>n) o=0;if(p<1) p=0;
      if(o==0 and p==0){
        cout<<-1<<" ";
      }else{
        if(o!=0 and p!=0){
          cout<<min(o-i,i-p)<<" ";
        }else if(o!=0){
          cout<<o-i<<" ";
        }else cout<<i-p<<" ";
      }

       
   }cout<<"\n";
   
   
}
int main(){
    ios;ll _=1;cin>>_;
    while (_--)solve();
    return 0;
}

可能会出错的特殊样例：

Case：

4
1
1
3
2 1 1
5
2 2 2 2 2
5
1 1 1 2 2 

Answer：

-1 
-1 1 2 
-1 -1 -1 -1 -1 
3 2 1 -1 2