Submission #863322

#	Time	Username	Problem	Language	Result	Execution time	Memory
863322		imarn	Meteors (POI11_met)	C++14	74 / 100	870 ms	46980 KiB

met

#include<bits/stdc++.h>
#define pb push_back
#define f first
#define ll long long
#define s second
using namespace std;
const int N=3e5+5;
ll fw[N];
void add(int i,ll amt){
    for(;i<N;i+=i&-i)fw[i]+=amt;
}
ll qr(int i,ll res=0){
    for(;i;i-=i&-i)res+=fw[i];
    return res;
}
vector<ll>g[N];
vector<ll>now[N];
int main(){
    ios_base::sync_with_stdio(false);cin.tie(0);
    int n,m;cin>>n>>m;
    ll a[n+1];
    ll b[m+1];
    ll c[n+1];
    for(int i=1;i<=m;i++)cin>>b[i],g[b[i]].pb(i);
    for(int i=1;i<=n;i++)cin>>c[i];
    int k;cin>>k;
    pair<ll,pair<int,int>>q[k+2];q[0]=q[k+1]={0,{1,m}};
    for(int i=1;i<=k;i++)cin>>q[i].s.f>>q[i].s.s>>q[i].f;
    int l[n+1],r[n+1];
    for(int i=1;i<=n;i++)l[i]=0,r[i]=k+1;
    while(1){
        memset(fw,0,sizeof fw);
        int cnt=0;
        for(int i=1;i<=n;i++){
            if(l[i]==r[i])continue;
            cnt++;now[(l[i]+r[i])/2].pb(i);
        }if(cnt==0)break;
        for(int i=0;i<=k+1;i++){
            if(q[i].s.f<=q[i].s.s)add(q[i].s.f,q[i].f),add(q[i].s.s+1,-q[i].f);
            else add(q[i].s.f,q[i].f),add(1,q[i].f),add(q[i].s.s+1,-q[i].f);
            for(auto it : now[i]){
                ll sum=0;
                for(auto ij : g[it])sum+=qr(ij);
                if(sum>=c[it])r[it]=i;
                else l[it]=i+1;
            }now[i].clear();
        }now[k+1].clear();
    }for(int i=1;i<=n;i++){
        if(r[i]>k)cout<<"NIE\n";
        else cout<<r[i]<<"\n";
    }
}

Compilation message (stderr)

met.cpp: In function 'int main()':
met.cpp:21:8: warning: unused variable 'a' [-Wunused-variable]
   21 |     ll a[n+1];
      |        ^

• 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: 0 / 13
• Subtask 8: 0 / 13