Submission #944144

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
944144	2024-03-12T09:00:24 Z	Warinchai	Meteors (POI11_met)	C++14	100 / 100	1562 ms	65536 KB

#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);
                    //assert(sum>=0);
                    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

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.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	8 ms	23640 KB	Output is correct
2	Correct	8 ms	23640 KB	Output is correct
3	Correct	7 ms	23644 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	7 ms	23644 KB	Output is correct
2	Correct	7 ms	23644 KB	Output is correct
3	Correct	7 ms	23644 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	57 ms	25032 KB	Output is correct
2	Correct	127 ms	26944 KB	Output is correct
3	Correct	84 ms	26632 KB	Output is correct

Subtask #4
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	118 ms	26212 KB	Output is correct
2	Correct	110 ms	26268 KB	Output is correct
3	Correct	127 ms	27080 KB	Output is correct
4	Correct	24 ms	25424 KB	Output is correct

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	87 ms	25712 KB	Output is correct
2	Correct	106 ms	27500 KB	Output is correct
3	Correct	113 ms	24476 KB	Output is correct
4	Correct	85 ms	27068 KB	Output is correct

Subtask #6
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	112 ms	24832 KB	Output is correct
2	Correct	112 ms	26164 KB	Output is correct
3	Correct	69 ms	25036 KB	Output is correct
4	Correct	116 ms	28108 KB	Output is correct

Subtask #7
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	899 ms	41400 KB	Output is correct
2	Correct	709 ms	30764 KB	Output is correct
3	Correct	608 ms	26828 KB	Output is correct
4	Correct	1562 ms	62256 KB	Output is correct

Subtask #8
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	838 ms	40044 KB	Output is correct
2	Correct	719 ms	30536 KB	Output is correct
3	Correct	492 ms	26604 KB	Output is correct
4	Correct	1535 ms	65536 KB	Output is correct