提出 #34483869

---

ソースコード 拡げる

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
#define rep(i,a,n) for (ll i=a;i<(ll)n;i++)

ll read(){ll r;scanf("%lld",&r);return r;} // read

char s[200010];
ll k;

vector<int> v;
vector<ll> pre = {0};

tuple<ll,ll,ll> calc(ll x,ll d){ // value = x, distance = d , return {<d的和, <d 的个数, =m的个数}
  // [x-d,x+d]
  int i  = lower_bound(v.begin(),v.end(),x    ) - v.begin();
  int i0 = lower_bound(v.begin(),v.end(),x-d  ) - v.begin();
  int i1 = lower_bound(v.begin(),v.end(),x-d+1) - v.begin();
  int i2 = lower_bound(v.begin(),v.end(),x+d  ) - v.begin();
  int i3 = lower_bound(v.begin(),v.end(),x+d+1) - v.begin();
  if(d == 0) return {0,0,i1-i0};
  // [i0 [i1 i  i2] i3]
  // [i1..i-1] + [i..i2-1]
  ll sum = (i-i1) * x - (pre[i] - pre[i1])
    + (pre[i2] - pre[i]) - (i2-i) * x;
  return {sum, i2-i1, (i1-i0)+(i3-i2) };
}

ll f(ll x){
  ll l = 0;
  ll r = max(x-v[0],v.back()-x)+1;
  while(l+1 < r){
    ll mid = (l+r)/2;
    (get<0>(calc(x, mid)) > k ? r : l) = mid;
  }
  auto [sum, c0, c1] = calc(x, l);
  return c0 + (l==0 ? c1 : min((k-sum)/l,c1));
}

int main(){
  scanf("%s",s);
  int n = strlen(s);
  k = read();
  rep(i,0,n) if(s[i] == 'Y'){ v.push_back(i-v.size()); }
  rep(i,0,v.size()) pre.push_back(pre.back() + v[i]);
  ll ans = 0;
  for(auto x:v) ans = max(ans, f(x));
  printf("%lld\n",ans);
  return 0;
}

提出情報

コンパイルエラー

./Main.cpp: In function ‘ll read()’:
./Main.cpp:7:21: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
    7 | ll read(){ll r;scanf("%lld",&r);return r;} // read
      |                ~~~~~^~~~~~~~~~~
./Main.cpp: In function ‘int main()’:
./Main.cpp:42:8: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   42 |   scanf("%s",s);
      |   ~~~~~^~~~~~~~

ジャッジ結果