Submission #761826

#	Time	Username	Problem	Language	Result	Execution time	Memory
761826		vjudge1	Meteors (POI11_met)	C++17	39 / 100	2853 ms	37704 KiB

met

#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 int B = 400;
 
ll lcm(ll a, ll b) { return (a / __gcd(a,b))*b; }
ll binpow(ll a, ll b, ll m) { ll res=1;a%=m; while(b>0){ if(b&1)res=(res*a)%m; a=(a*a)%m; b/=2; } return res%m;}
 
void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }

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) {
				int start = max(1ll, ansblock[x] * B - B + 1), final = i;
				for(int 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(); }
}

Compilation message (stderr)

met.cpp: In function 'void Freopen(std::string)':
met.cpp:31:34: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 | void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }
      |                           ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
met.cpp:31:76: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 | void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }
      |                                                                     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 12
• Subtask 2: 0 / 12
• Subtask 3: 0 / 12
• Subtask 4: 0 / 12
• Subtask 5: 13 / 13
• Subtask 6: 0 / 13
• Subtask 7: 13 / 13
• Subtask 8: 13 / 13