Submission #944137

#	Time	Username	Problem	Language	Result	Execution time	Memory
944137		Warinchai	Meteors (POI11_met)	C++14	100 / 100	1451 ms	65536 KiB

met

#include<bits/stdc++.h>
//#define int long long
using namespace std;
int own[350005];
vector<int>states[350005];
int least[350005];
struct fenwick{
    long long info[350005];
    void upd(int x,long long val){
        for(int i=x;i<=350001;i+=i&-i)info[i]+=val;
    }
    long long fval(int x){
        long long sum=0;
        for(int i=x;i>0;i-=i&-i)sum+=info[i];
        return sum;
    }
    void clear(){
        for(int i=1;i<=350001;i++)info[i]=0;
    }
}tr;
int st[350005];
int en[350005];
struct meteor{
    int l,r,a;
    meteor(int x,int y,int z){
        l=x,r=y,a=z;
    }
};
vector<meteor>events;
vector<int>query[350005];
int32_t main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int n,m;cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>own[i];
        states[own[i]].push_back(i);
    }
    for(int i=1;i<=n;i++){
        cin>>least[i];
    }
    int k;cin>>k;
    long long a,b,c,md,cur,l,r,sum,x;
    for(int i=1;i<=k;i++){
        a,b,c;cin>>a>>b>>c;
        events.push_back(meteor(a,b,c));
    }
    for(int i=1;i<=n;i++){
        st[i]=1;en[i]=k+1;
        md=(st[i]+en[i])/2;
        query[md].push_back(i);
    }
    //cout<<"work\n";
    while(1){
        cur=0;
        for(auto x:events){
            cur++;
            //cout<<"EVENT:"<<cur<<"\n";
            l=x.l;
            r=x.r;
            if(l<=r)tr.upd(l,x.a),tr.upd(r+1,x.a*-1);
            else tr.upd(l,x.a),tr.upd(m+1,x.a*-1),tr.upd(1,x.a),tr.upd(r+1,x.a*-1);
            for(auto y:query[cur]){
                //cerr<<"STATE:"<<y<<"\n";
                sum=0;
                for(auto z:states[y]){
                    //cerr<<"STATION:"<<z<<" "<<tr.fval(1,m,1,z)<<"\n";
                    sum+=tr.fval(z);
                    if(sum>=least[y])break;
                }
                if(sum>=least[y]){
                    //cerr<<"ENOUGH\n";
                    en[y]=cur;
                }else{
                    //cerr<<"NOT ENOUGH\n";
                    st[y]=cur+1;
                }
            }
            query[cur].clear();
        }
        tr.clear();
        int check=1;
        for(int i=1;i<=n;i++){
            if(st[i]!=en[i]){
                check=0;
                md=(st[i]+en[i])/2;
                query[md].push_back(i);
            }
        }
        if(check)break;
    }
    for(int i=1;i<=n;i++){
        x=st[i];
        if(x==k+1)cout<<"NIE\n";
        else cout<<x<<"\n";
    }
}

Compilation message (stderr)

met.cpp: In function 'int32_t main()':
met.cpp:45:9: warning: left operand of comma operator has no effect [-Wunused-value]
   45 |         a,b,c;cin>>a>>b>>c;
      |         ^
met.cpp:45:13: warning: right operand of comma operator has no effect [-Wunused-value]
   45 |         a,b,c;cin>>a>>b>>c;
      |             ^
met.cpp:45:14: warning: right operand of comma operator has no effect [-Wunused-value]
   45 |         a,b,c;cin>>a>>b>>c;
      |              ^

• Subtask 1: 12 / 12
• Subtask 2: 12 / 12
• Subtask 3: 12 / 12
• Subtask 4: 12 / 12
• Subtask 5: 13 / 13
• Subtask 6: 13 / 13
• Subtask 7: 13 / 13
• Subtask 8: 13 / 13