Submission #488950

#	Time	Username	Problem	Language	Result	Execution time	Memory
488950		MilosMilutinovic	Meteors (POI11_met)	C++14	74 / 100	1972 ms	65540 KiB

met

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
typedef double db;
mt19937 mrand(random_device{}()); 
const ll mod=1000000007;
const ll mod2=998244353;
int rnd(int x) { return mrand() % x;}
ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}
// head

const int N=301000;
int n,m,q,o[N],p[N],l[N],r[N],a[N],up[N],lo[N],ans[N];
ll tot[N];
VI qv[N],qs[N];

struct ds {
	ll c[N];
	ds() {
		rep(i,0,N) c[i]=0;
	}
	void modify(int x,int y) {
		for (;x<N;x+=x&-x) c[x]+=y;
	}
	ll query(int x) {
		ll s=0;
		for (;x;x-=x&-x) s+=c[x];
		return s;
	}
};

signed main() {
	scanf("%lld%lld",&n,&m);
	rep(i,1,m+1) scanf("%lld",o+i),qv[o[i]].pb(i);
	rep(i,1,n+1) scanf("%lld",p+i);
	scanf("%lld",&q);
	rep(i,1,q+1) scanf("%lld%lld%lld",l+i,r+i,a+i);
	rep(i,1,n+1) lo[i]=1,up[i]=q,ans[i]=q+1;
	rep(it,0,50) {
		rep(i,1,n+1) tot[i]=0;
		rep(i,1,q+1) qs[i].clear();
		rep(i,1,n+1) if (lo[i]<=up[i])
			qs[(lo[i]+up[i])/2].pb(i);
		ds fenw;
		rep(i,1,q+1) {
			if (l[i]<=r[i]) {
				fenw.modify(l[i],a[i]);
				fenw.modify(r[i]+1,-a[i]);
			} else {
				fenw.modify(l[i],a[i]);
				fenw.modify(1,a[i]);
				fenw.modify(r[i]+1,-a[i]);
			}
			for (auto x:qs[i]) for (auto y:qv[x]) if (tot[x]<p[x]) 
				tot[x]+=fenw.query(y);
		}
		rep(i,1,n+1) {
			if (lo[i]>up[i]) continue;
			int md=(lo[i]+up[i])/2;
			if (tot[i]>=p[i]) ans[i]=md,up[i]=md-1;
			else lo[i]=md+1;
		}
	}
	rep(i,1,n+1) {
		if (ans[i]>q) puts("NIE");
		else printf("%lld\n",ans[i]);
	}
}

Compilation message (stderr)

met.cpp: In function 'int main()':
met.cpp:45:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   45 |  scanf("%lld%lld",&n,&m);
      |  ~~~~~^~~~~~~~~~~~~~~~~~
met.cpp:46:20: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   46 |  rep(i,1,m+1) scanf("%lld",o+i),qv[o[i]].pb(i);
      |               ~~~~~^~~~~~~~~~~~
met.cpp:47:20: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   47 |  rep(i,1,n+1) scanf("%lld",p+i);
      |               ~~~~~^~~~~~~~~~~~
met.cpp:48:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   48 |  scanf("%lld",&q);
      |  ~~~~~^~~~~~~~~~~
met.cpp:49:20: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   49 |  rep(i,1,q+1) scanf("%lld%lld%lld",l+i,r+i,a+i);
      |               ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~

• 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