Submission #761839

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
761839	2023-06-20T10:23:11 Z	Baizho	Meteors (POI11_met)	C++14	39 / 100	2759 ms	37752 KB

#include <bits/stdc++.h>
using namespace std;
 
// #pragma comment(linker, "/stack:2000000000")
// #pragma GCC optimize("Ofast,unroll-loops,fast-math,O3")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")

 
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll,ll> pll;
 
#define sz size()
#define ff first
#define ss second
#define all(a) a.begin(),a.end()
#define pb push_back

const int mod = ll(1e9)+7; //(b + (a%b)) % b (to mod -1%(10^9+7) correctly in c++ its -1 but its suppose to be 10^9+6
const int MOD = 998244353;  // (a%mod)*(binpow(b,mod-2,mod) = (a/b)%mod
const int N = ll(4e5)+10;
const long long inf = 1e18;
//const int inf = 1e9;
const long double eps = 1e-15L;
const ll B = 400;
 

ll n, m, o[N], p[N], k, ansblock[N], ans[N], ql[N], qr[N], qa[N];
ll scan[N], origp[N];
vector<ll>pos[N];

void Baizho() {
	cin>>n>>m;
	for(int i = 1; i<=m; i++) {
		cin>>o[i];
		ans[o[i]] = ansblock[o[i]] = -inf;
		pos[o[i]].pb(i);
	}
	for(int i = 1; i<=n; i++) {
		cin>>p[i];
		origp[i] = p[i];
	}
	cin>>k;
	for(int i = 1; i<=k; i++) {
		ll l, r, a;
		cin>>l>>r>>a;
		ql[i] = l;
		qr[i] = r;
		qa[i] = a;
		if(l > r) {
			scan[l] += a;
			scan[m + 1] -= a;
			scan[1] += a;
			scan[r + 1] -= a;
		}
		else {
			scan[l] += a;
			scan[r + 1] -= a;
		}
		
		if(i == k || i % B == 0) {
			vector<ll> vec;
			for(int j = 1; j<=m; j++) {
				scan[j] += scan[j - 1];
				p[o[j]] -= scan[j];
				if(p[o[j]] <= 0 && ansblock[o[j]] == -inf) {
					ansblock[o[j]] = (i + B - 1) / B;
					vec.pb(o[j]);
				}
			}
			
			for(auto x : vec) {
				ll start = max(1ll, ansblock[x] * B - B + 1), final = i;
				for(ll h = start; h <= final; h++) {
					ll L, R;
					if(ql[h] > qr[h]) {
						L = ql[h], R = m;
						ll cnt = upper_bound(all(pos[x]), R) - pos[x].begin() - (lower_bound(all(pos[x]), L) - pos[x].begin());
						origp[x] -= cnt * qa[h];
						if(origp[x] <= 0 && ans[x] == -inf) {
							ans[x] = h;
							break;
						}
						L = 1, R = qr[h];
						cnt = upper_bound(all(pos[x]), R) - pos[x].begin() - (lower_bound(all(pos[x]), L) - pos[x].begin());
						origp[x] -= cnt * qa[h];
						if(origp[x] <= 0 && ans[x] == -inf) {
							ans[x] = h;
							break;
						}
 					}
 					else {
 						L = ql[h], R = qr[h];
						ll cnt = upper_bound(all(pos[x]), R) - pos[x].begin() - (lower_bound(all(pos[x]), L) - pos[x].begin());
						origp[x] -= cnt * qa[h];
						if(origp[x] <= 0 && ans[x] == -inf) {
							ans[x] = h;
							break;
						} 
					}
				}
			}
			for(int j = 0; j<=m + 1; j++) {
				scan[j] = 0;
				if(1 <= j && j <= m) origp[o[j]] = p[o[j]];
			}
		}
	}
	for(int i = 1; i<=n; i++) {
		if(ans[i] == -inf) cout<<"NIE\n";
		else cout<<ans[i]<<"\n";
	}
}
 
int main() {
 
//     Freopen("guard");
    ios_base::sync_with_stdio(false);   
    cin.tie(0);cout.tie(0); 
    
   
    int ttt = 1;
//    cin>>ttt;

    for(int i=1; i<=ttt; i++) {Baizho(); }
}

Subtask #1
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	5 ms	9812 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #2
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	5 ms	9812 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #3
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Incorrect	36 ms	12316 KB	Output isn't correct
2	Halted	0 ms	0 KB	-

Subtask #4
0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	74 ms	12692 KB	Output is correct
2	Correct	86 ms	12748 KB	Output is correct
3	Incorrect	89 ms	13584 KB	Output isn't correct
4	Halted	0 ms	0 KB	-

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	43 ms	12600 KB	Output is correct
2	Correct	65 ms	13400 KB	Output is correct
3	Correct	14 ms	10964 KB	Output is correct
4	Correct	83 ms	13052 KB	Output is correct

Subtask #6
0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	45 ms	12240 KB	Output is correct
2	Incorrect	71 ms	12724 KB	Output isn't correct
3	Halted	0 ms	0 KB	-

Subtask #7
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1622 ms	31084 KB	Output is correct
2	Correct	812 ms	23960 KB	Output is correct
3	Correct	42 ms	15648 KB	Output is correct
4	Correct	2477 ms	35756 KB	Output is correct

Subtask #8
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	1552 ms	29940 KB	Output is correct
2	Correct	809 ms	23916 KB	Output is correct
3	Correct	40 ms	14500 KB	Output is correct
4	Correct	2759 ms	37752 KB	Output is correct

Submission #761839

Compilation message