답안 #142662

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
142662 2019-08-10T10:11:45 Z Bodo171 가로등 (APIO19_street_lamps) C++14
0 / 100
657 ms 1548 KB
#include <iostream>
#include <fstream>
#include <set>
using namespace std;
const int nmax=1005;
struct pct
{
    int x,y,val;
}nl;
int n,i,q,j,poz,a,b;
string ini;
bool operator <(pct unu,pct doi)
{
    return unu.x<doi.x;
}
bool operator !=(pct unu,pct doi)
{
    return (unu.x!=doi.x||unu.y!=doi.y||unu.val!=doi.val);
}
struct node
{
    node *ls,*rs;
    int sum;
    node()
    {
        ls=0;rs=0;
        sum=0;
    }
};
set<pct> s;
set<pct>::iterator it;
string tip;
struct aint
{
    node *arb[4*nmax];
    int nr,stX,drX,stY,drY;
    void uY(node *nod,int l,int r,int poz,int val)
    {
        nod->sum+=poz;
        if(l==r) return;
        int m=(l+r)/2;
        if(poz<=m)
        {
            if(!nod->ls)
                nod->ls=new node;
            uY(nod->ls,l,m,poz,val);
        }
        else
        {
            if(!nod->rs)
                nod->rs=new node;
            uY(nod->rs,m+1,r,poz,val);
        }
    }
    void qY(node *nod,int l,int r)
    {
        if(stY<=l&&r<=drY)
        {
            ret+=nod->sum;
            return;
        }
        int m=(l+r)/2;
        if(stY<=m&&nod->ls) qY(nod->ls,l,m);
        if(m<drY&&nod->rs)  qY(nod->rs,m+1,r);
    }
    void update(int nod,int l,int r,int x,int y,int val)
    {
        uY(arb[nod],1,n,y,val);
        if(l==r) return;
        int m=(l+r)/2;
        if(x<=m) update(2*nod,l,m,x,y,val);
        else update(2*nod+1,m+1,r,x,y,val);
    }
    int ret;
    void query(int nod,int l,int r)
    {
        if(stX<=l&&r<=drX)
        {
            qY(arb[nod],1,n);
            return;
        }
        int m=(l+r)/2;
        if(stX<=m) query(2*nod,l,m);
        if(m<drX)  query(2*nod+1,m+1,r);
    }
    int Q(int x,int y)
    {
        ret=0;
        stX=1;drX=x;stY=y;drY=n;
        query(1,1,n);
        return ret;
    }
    void build(int nod,int l,int r)
    {
        arb[nod]=new node;
        if(l==r) return;
        int m=(l+r)/2;
        build(2*nod,l,m);
        build(2*nod+1,m+1,r);
    }
}A;
void ins(int poz)
{
    pct st=nl,dr=nl,act;
    it=s.lower_bound({poz,0,0});
    if((*it).x==poz+1)
        dr=(*it);
    if(it!=s.begin())
    {
        it--;
        if((*it).y==poz-1)
            st=(*it);
    }
    act.val=i;act.x=act.y=poz;
    if(st!=nl)
    {
        A.update(1,1,n,st.x,st.y,i-st.val);
        act.x=st.x;
        s.erase(st);
    }
    if(dr!=nl)
    {
        A.update(1,1,n,dr.x,dr.y,i-dr.val);
        act.y=dr.y;
        s.erase(dr);
    }
    s.insert(act);
    return;
}
void del(int poz)
{
    pct st=nl,dr=nl,act;
    it=s.lower_bound({poz+1,0,0});
    it--;//mereu se poate
    act=(*it);
    A.update(1,1,n,act.x,act.y,i-act.val);
    s.erase(act);
    if(poz>act.x)
    {
        st={act.x,poz-1,i};
        s.insert(st);
    }
    if(poz<act.y)
    {
        dr={poz+1,act.y,i};
        s.insert(dr);
    }

}
int main()
{
    //freopen("data.in","r",stdin);
    ios_base::sync_with_stdio(false);
    cin>>n>>q;
    cin>>ini;
    ini=" "+ini;
    for(i=1;i<ini.size();i++)
    {
        if(ini[i]=='1')
        {
            j=i;
            while(j<ini.size()&&ini[j]=='1')
                j++;
            j--;
            s.insert({i,j,0});
            i=j;
        }
    }
    A.build(1,1,n);
    for(i=1;i<=q;i++)
    {
        cin>>tip;
        if(tip=="toggle")
        {
            cin>>poz;
            if(ini[poz]=='1')
            {
                ini[poz]='0';
                del(poz);
            }
            else
            {
                ini[poz]='1';
                ins(poz);
            }
        }
        else
        {
            int ans;
            cin>>a>>b;
            it=s.lower_bound({b,0,0});
            ans=0;
            if(it!=s.begin())
            {
                it--;
                if((*it).x<=a&&(*it).y>=b-1) ans=i-(*it).val;
            }
            ans+=A.Q(a,b-1);
            cout<<ans<<'\n';
        }
    }
    return 0;
}

Compilation message

street_lamps.cpp: In function 'int main()':
street_lamps.cpp:157:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(i=1;i<ini.size();i++)
             ~^~~~~~~~~~~
street_lamps.cpp:162:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             while(j<ini.size()&&ini[j]=='1')
                   ~^~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 376 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 657 ms 1548 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 1528 KB Output is correct
2 Incorrect 6 ms 1400 KB Output isn't correct
3 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 6 ms 504 KB Output is correct
2 Incorrect 5 ms 888 KB Output isn't correct
3 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 2 ms 376 KB Output isn't correct
2 Halted 0 ms 0 KB -