Submission #1280919

#	Time	Username	Problem	Language	Result	Execution time	Memory
1280919		StefanSebez	Street Lamps (APIO19_street_lamps)	C++20	20 / 100	4135 ms	185148 KiB

street_lamps

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ll long long
#define ld long double
const int N=3e5+50;
void Unique(vector<int>&a){sort(a.begin(),a.end());a.resize(unique(a.begin(),a.end())-a.begin());}
int a[N],n,q;
vector<array<int,3>>Qs;
struct BIT2D{
    vector<int>T[N+10],vals[N+10];
    void Init(){for(int i=0;i<=N;i++){vals[i].pb(0);Unique(vals[i]);T[i].resize(vals[i].size());}}
    void Dodaj(int i,int j){for(;i<=N;i+=i&-i) vals[i].pb(j);}
    void Update(int i,int j,int x){for(;i<=N;i+=i&-i) Update1(i,j,x);}
    void Update1(int i,int j,int x){for(int k=upper_bound(vals[i].begin(),vals[i].end(),j)-vals[i].begin()-1;k<vals[i].size();k+=k&-k)T[i][k]+=x;}
    int Get(int i,int j){int x=0;for(;i>=1;i-=i&-i)x+=Get1(i,j);return x;}
    int Get1(int i,int j){int x=0;for(int k=upper_bound(vals[i].begin(),vals[i].end(),j)-vals[i].begin()-1;k>=1;k-=k&-k)x+=T[i][k];return x;}
    int Get(int l,int L,int r,int R){return Get(L,R)-Get(l-1,R)-Get(L,r-1)+Get(l-1,r-1);}
}bt[3];
struct BIT{
    int T[N+10];
    void Update(int i,int x){for(;i<=N;i+=i&-i)T[i]+=x;}
    int Get(int i){int x=0;for(;i>=1;i-=i&-i)x+=T[i];return x;}
    int Get(int l,int r){return Get(r)-Get(l-1);}
    void Clear(){for(int i=0;i<=N;i++) T[i]=0;}
}BT;
void Toggle(int x,int I,bool dodaj){
    int val=BT.Get(x,x);
    BT.Update(x,1-val);
    //for(int i=1;i<=n;i++) printf("%i ",BT.Get(i,i));printf("\n");
    int l=x+1,r=n,desno=x;
    while(l<=r){
        int mid=l+r>>1;
        //printf("%i %i %i: %i %i\n",l,mid,r,BT.Get(x+1,mid),mid-(x+1)+1);
        if(BT.Get(x+1,mid)==mid-(x+1)+1) l=mid+1,desno=mid;
        else r=mid-1;
    }
    l=1,r=x-1;int levo=x;
    while(l<=r){
        int mid=l+r>>1;
        if(BT.Get(mid,x-1)==(x-1)-mid+1) r=mid-1,levo=mid;
        else l=mid+1;
    }
    //printf("%i %i %i\n",levo,x,desno);
    if(dodaj){
        if(val==1){
            if(levo<x) for(int i=0;i<3;i++)bt[i].Dodaj(levo,x-1);
            if(x<desno) for(int i=0;i<3;i++)bt[i].Dodaj(x+1,desno);
        }
        else for(int i=0;i<3;i++)bt[i].Dodaj(levo,desno);
    }
    else{
        if(val==1){
            bt[0].Update(levo,desno,-1);
            int tajm=bt[1].Get(levo,levo,desno,desno);
            bt[1].Update(levo,desno,-tajm);
            bt[2].Update(levo,desno,I-tajm);
            if(levo<x){
                bt[0].Update(levo,x-1,1);
                bt[1].Update(levo,x-1,I);
            }
            if(x<desno){
                bt[0].Update(x+1,desno,1);
                bt[1].Update(x+1,desno,I);
            }
        }
        else{
            //zavrsi levo,x-1
            if(levo<=x-1){
                bt[0].Update(levo,x-1,-1);
                int tajm=bt[1].Get(levo,levo,x-1,x-1);
                bt[1].Update(levo,x-1,-tajm);
                bt[2].Update(levo,x-1,I-tajm);
            }
            //zavrsi x+1,desno
            if(x+1<=desno){
                bt[0].Update(x+1,desno,-1);
                int tajm=bt[1].Get(x+1,x+1,desno,desno);
                bt[1].Update(x+1,desno,-tajm);
                bt[2].Update(x+1,desno,I-tajm);
            }
            //dodaj levo,desno
            bt[0].Update(levo,desno,1);
            bt[1].Update(levo,desno,I);
        }
    }
}
int main(){
    scanf("%i%i",&n,&q);
    string s;cin>>s;
    for(int i=1;i<=n;i++) a[i]=s[i-1]-'0';
    for(int i=1;i<=n;i++) BT.Update(i,a[i]);
    for(int i=1;i<=n;){
        if(a[i]==0){i++;continue;}
        int j=i;
        while(j<=n&&a[j]==1) j++;
        for(int k=0;k<3;k++)bt[k].Dodaj(i,j-1);
        i=j;
    }
    for(int I=1;I<=q;I++){
        string type;cin>>type;
        if(type=="toggle"){
            int x;scanf("%i",&x);
            Qs.pb({0,x,0});
            Toggle(x,I,true);
        }
        else{
            int l,r;scanf("%i%i",&l,&r);
            r--;
            Qs.pb({1,l,r});
        }
    }
    for(int i=0;i<3;i++)bt[i].Init();
    BT.Clear();
    for(int i=1;i<=n;i++) BT.Update(i,a[i]);
    for(int i=1;i<=n;){
        if(a[i]==0){i++;continue;}
        int j=i;
        while(j<=n&&a[j]==1) j++;
        bt[0].Update(i,j-1,1);
        i=j;
    }
    for(int I=1;I<=q;I++){
        auto [type,l,r]=Qs[I-1];
        if(type==0) Toggle(l,I,false);
        else{
            int res=bt[2].Get(1,l,r,n);
            res+=I*bt[0].Get(1,l,r,n)-bt[1].Get(1,l,r,n);
            //printf("%i %i %i\n",bt[0].Get(1,l,r,n),bt[1].Get(1,l,r,n),bt[2].Get(1,l,r,n));
            printf("%i\n",res);
        }
    }
    return 0;
}

Compilation message (stderr)

street_lamps.cpp: In function 'int main()':
street_lamps.cpp:91:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   91 |     scanf("%i%i",&n,&q);
      |     ~~~~~^~~~~~~~~~~~~~
street_lamps.cpp:105:24: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  105 |             int x;scanf("%i",&x);
      |                   ~~~~~^~~~~~~~~
street_lamps.cpp:110:26: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
  110 |             int l,r;scanf("%i%i",&l,&r);
      |                     ~~~~~^~~~~~~~~~~~~~

• Subtask 1: 0 / 20
• Subtask 2: 0 / 20
• Subtask 3: 20 / 20
• Subtask 4: 0 / 20
• Subtask 5: 0 / 20