oj.uz

답안 #919462

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
919462 2024-01-31T21:41:34 Z noyancanturk Meteors (POI11_met) C++17
74 / 100
 628 ms 38388 KB 
#ifndef Local
    #pragma GCC optimize("O3,unroll-loops")
    const int lim=3e5+100;
#else
    const int lim=1e3+100;
#endif
    
#include "bits/stdc++.h"
using namespace std;
    
#define int int64_t
#define pb push_back
    
const int mod=1e9+7;
using pii=pair<int,int>;

int tree[lim];
struct fenwick{
    int n;
    fenwick(int n):n(n){
        for(int i=0;i<=n;i++)tree[i]=0;
    }
    void update(int p,int val){
        while(p<=n){
            tree[p]+=val;
            p+=p&-p;
        }
    }
    void update(int l,int r,int val){
        update(l,val),update(r+1,-val);
    }
    int query(int p){
        int ans=0;
        while(p){
            ans+=tree[p];
            p-=p&-p;
        }
        return ans;
    }
};

struct bs{
    signed l,r,i;
};

inline void solve(){
    int n,m;
    cin>>n>>m;
    vector<signed>v[n+1];
    for(int i=1;i<=m;i++){
        int tem;
        cin>>tem;
        v[tem].pb(i);
    }
    signed goal[n+1];
    for(int i=1;i<=n;i++){
        cin>>goal[i];
    }
    int k;
    cin>>k;
    bs all[k+1];
    for(int i=1;i<=k;i++){
        cin>>all[i].l>>all[i].r>>all[i].i;
    }
    vector<int>qq[k+1];
    for(int i=1;i<=n;i++){
        qq[(1+k)>>1].pb(i);
    }
    signed l[k+1],r[k+1];
    l[(1+k)>>1]=1,r[(1+k)>>1]=k;
    signed ans[n+1];
    memset(ans,-1,sizeof(ans));
    bool ok=1;
    while(ok){
        ok=0;
        fenwick st(m+10);
        for(int t=1;t<=k;t++){
            if(all[t].l<=all[t].r)st.update(all[t].l,all[t].r,all[t].i);
            else {
                st.update(all[t].l,m,all[t].i);
                st.update(1,all[t].r,all[t].i);
            }
            while(qq[t].size()){
                int now=qq[t].back();
                qq[t].pop_back();
                //cerr<<now.l<<" "<<now.r<<" "<<now.i<<"\n\n";
                int tot=0;
                for(int j:v[now]){
                    tot+=st.query(j);
                }
                //cerr<<mid<<" "<<tot<<"\n\n";
                int mid=(l[t]+r[t])>>1;
                if(goal[now]<=tot){
                    ans[now]=mid;
                    if(l[t]<=mid-1){
                        qq[(l[t]+mid-1)>>1].pb(now);
                        l[(l[t]+mid-1)>>1]=l[t];
                        r[(l[t]+mid-1)>>1]=mid-1;
                        ok=1;
                    }
                }else{
                    if(mid+1<=r[t]){
                        qq[(mid+1+r[t])>>1].pb(now);
                        l[(mid+1+r[t])>>1]=mid+1;
                        r[(mid+1+r[t])>>1]=r[t];
                        ok=1;
                    }
                }
            }
        }
    }
    for(int i=1;i<=n;i++){
        if(ans[i]==-1)cout<<"NIE\n";
        else cout<<ans[i]<<"\n";
    }
}
    
signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
#ifdef Local  
    freopen(".in","r",stdin);
    freopen(".out","w",stdout);
#else

#endif
    int t=1;
    //cin>>t;
    while (t--)
    {
        solve();
    }
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct

Subtask #2
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 2 ms 600 KB Output is correct
3 Correct 1 ms 600 KB Output is correct

Subtask #3
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 48 ms 5976 KB Output is correct
2 Correct 78 ms 9600 KB Output is correct
3 Correct 67 ms 8324 KB Output is correct

Subtask #4
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 68 ms 7504 KB Output is correct
2 Correct 75 ms 7640 KB Output is correct
3 Correct 83 ms 9760 KB Output is correct
4 Correct 22 ms 5760 KB Output is correct

Subtask #5
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 54 ms 6224 KB Output is correct
2 Correct 63 ms 10300 KB Output is correct
3 Correct 29 ms 2652 KB Output is correct
4 Correct 66 ms 8908 KB Output is correct

Subtask #6
13.0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 62 ms 5200 KB Output is correct
2 Correct 71 ms 7508 KB Output is correct
3 Correct 49 ms 5464 KB Output is correct
4 Correct 83 ms 11472 KB Output is correct

Subtask #7
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 510 ms 38388 KB Output is correct
2 Incorrect 628 ms 17024 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 515 ms 36640 KB Output is correct
2 Incorrect 616 ms 16840 KB Output isn't correct
3 Halted 0 ms 0 KB -