提出 #38686072

---

ソースコード 拡げる

#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;}

int score[200010];
int quota[200010];
vector<array<int,3> >Q;
vector<ll> xval;
ll init[400010]; // {quota sum}

array<ll,2> operator+(array<ll,2>l,array<ll,2>r){ return {l[0]+r[0],l[1]+r[1]}; }
array<ll,2> operator-(array<ll,2>l,array<ll,2>r){ return {l[0]-r[0],l[1]-r[1]}; }

class Fenwick{
public:
  vector<array<ll,2> > qs; // quota, score sum
  vector<array<ll,2> > point; // quota, score sum
  Fenwick(int sz){ // 0-index
    qs.resize(sz);
    point.resize(sz);
  }
  void add(int x,int inc){
    array<ll,2> diff = {inc, inc*xval[x]};
    point[x] = point[x] + diff;
    for(x+=1;x<=(int)qs.size();x+=(x&-x)){
      qs[x-1] = qs[x-1] + diff;
    }
  }

  ll query(int cnt){
    int x = qs.size();
    ll res = 0;
    for(;x;){
      if(qs[x-1][0] <= cnt){
        cnt -= qs[x-1][0];
        res += qs[x-1][1];
        x-=(x&-x);
      }else{
        if(point[x-1][0] <= cnt){
          cnt -= point[x-1][0];
          res += point[x-1][1];
          x-=1;
        }else{
          return res += cnt*xval[x-1];
        }
      }
    }
    return cnt==0?res:-1;
  }
};

int main(){
  int n=read();
  rep(i,0,n) {
    score[i]=read(); // score
    quota[i]=read(); // quota
    xval.push_back(score[i]);
  }
  int q=read();
  rep(i,0,q){
    int op=read();
    int x=read();
    if(op==1){
      int y=read();
      xval.push_back(y);
      Q.push_back({op,x-1,y});
    }else if(op==2){
      int y=read();
      Q.push_back({op,x-1,y});
    }else if(op==3){
      Q.push_back({op,x,0});
    }
  }
  sort(begin(xval),end(xval));
  xval.erase(unique(begin(xval),end(xval)),end(xval));
  auto idx=[&](int x){return lower_bound(begin(xval),end(xval),x)-begin(xval);};
  Fenwick fw(xval.size());
  rep(i,0,n) fw.add(idx(score[i]), quota[i]);
  for(auto [op,x,y]:Q){
    if(op==1){
      fw.add(idx(score[x]),-quota[x]);
      score[x] = y;
      fw.add(idx(score[x]),quota[x]);
    }else if(op==2){
      fw.add(idx(score[x]),y-quota[x]);
      quota[x] = y;
    }else if(op==3){
      printf("%lld\n", fw.query(x));
    }
  }
  return 0;
}

提出情報

コンパイルエラー

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

ジャッジ結果