Submission #890455

#	Time	Username	Problem	Language	Result	Execution time	Memory
890455		vjudge1	새로운 문제 (POI11_met)	C++17	0 / 100	2175 ms	39992 KiB

met

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

//#pragma GCC optimization("g", on)
//#pragma GCC optimization("03")
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("inline")
//#pragma GCC optimize("-fgcse,-fgcse-lm")
//#pragma GCC optimize("-ftree-pre,-ftree-vrp")
//#pragma GCC optimize("-ffast-math")
//#pragma GCC optimize("-fipa-sra")
//#pragma GCC optimize("-fpeephole2")
//#pragma GCC optimize("-fsched-spec")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
//#pragma GCC optimize("unroll-loops")

#define aboba ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define br break;
#define sp " "
#define en "\n"
#define pb push_back
#define sz(a) int(a.size())
#define bg begin()
#define ed end()
#define in insert
#define ss second
#define ff first
#define setp(a) cout << fixed; cout << setprecision(a)
#define all(v) v.begin(), v.end()
 
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef double db;
typedef tree<
    long long,
    null_type,
    less_equal<long long>,
    rb_tree_tag,
    tree_order_statistics_node_update> orset;

void freopen(string s) { freopen((s + ".in").c_str(), "r", stdin); freopen((s + ".out").c_str(), "w", stdout); }
ll bm(ll x, ll y, ll z) { ll res = 0; while (y) { if (y & 1) (res += x) %= z; (x += x) %= z; y >>= 1; } return res; }
ll bp(ll x, ll y, ll z) { ll res = 1; x %= z; while (y) { if (y & 1) { (res *= x) %= z; y--; } (x *= x) %= z; y >>= 1; } return res; }
// C(n, k) = ((fact[n] * bp(fact[k], mod - 2)) % mod * bp(fact[n - k], mod - 2)) % mod;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
ll lcm(ll a, ll b) { return (a / __gcd(a, b)) * b; }

const ll N = 3e5 + 11;
const ll inf = 1e18 + 7;
ll tt = 1;
ll n, m, a[N], b[N];
ll ans[N], sum[N], d[N];
vector<ll> g[N], gr[N];
ll l[N], r[N], x[N];
bool ok[N];

void solve() {
	cin >> n >> m;
	for (int i = 1;i <= m;i++) {
		cin >> a[i];
		g[a[i]].pb(i);
	}
	for (int i = 1;i <= n;i++) cin >> b[i], ans[i] = -1;
	ll k; cin >> k;
	ll B = min(k, 600ll);
	for (ll i = 1;i <= k;i++) {
		cin >> l[i] >> r[i] >> x[i];
		if (l[i] > r[i]) {
			d[1] += x[i];
			d[r[i] + 1] -= x[i];
			d[l[i]] += x[i];
			d[m + 1] -= x[i];
		} else {
			d[l[i]] += x[i];
			d[r[i] + 1] -= x[i];
		}
		if (i % B == 0 || i == k) {
			ll tm = 0;
			for (int j = 1;j <= m;j++) {
				tm += d[j];
				sum[a[j]] += tm;
//				d[j] = 0;	
			}
			for (int j = 1;j <= n;j++) {
//				cout << sum[j] << sp << j << sp << i << en;
				if (sum[j] >= b[j] && ans[j] == -1) {
					ans[j] = (i - 1) / B;
//					cout << j << en;
					gr[(i - 1) / B].pb(j);
				}
			}
		}
	}
	for (int i = 1;i <= m;i++) d[i] = 0;
	for (int i = 1;i <= n;i++) sum[i] = 0;
	for (ll i = 1;i <= k;i++) {
		for (auto to : gr[(i - 1) / B]) {
			if (ok[to]) continue;
			if (g[to].empty()) continue;
			if (l[i] > r[i]) {
				ll pos = lower_bound(all(g[to]), r[i]) - g[to].bg;
//				cout << pos << sp << i << sp << to << en;
				sum[to] += x[i] * (sz(g[to]) - pos);
				pos = upper_bound(all(g[to]), l[i]) - g[to].bg;
				pos--;
				sum[to] += x[i] * pos;
			} else {
				ll pos = lower_bound(all(g[to]), l[i]) - g[to].bg;
				ll pos1 = upper_bound(all(g[to]), r[i]) - g[to].bg;
				pos1--;
				sum[to] += x[i] * (pos1 - pos + 1);
			}
			if (sum[to] >= b[to]) {
				ok[to] = 1;
				ans[to] = i;
			}
		}
		if (l[i] > r[i]) {
			d[1] += x[i];
			d[r[i] + 1] -= x[i];
			d[l[i]] += x[i];
			d[m + 1] -= x[i];
		} else {
			d[l[i]] += x[i];
			d[r[i] + 1] -= x[i];
		}
		if (i % B == 0 || i == k) {
			ll tm = 0;
			for (int j = 1;j <= m;j++) {
				tm += d[j];
				sum[a[j]] += tm;
//				d[j] = 0;
			}
		}
	} 
	for (int i = 1;i <= n;i++) {
		if (ans[i] == -1) cout << "NIE" << en;
		else cout << ans[i] << en;
	}
}  
   
int main() {      
    aboba    
//    freopen("haircut"); 
//    precalc();
//    cin >> tt;
    for (int i = 1;i <= tt;i++) {			
		solve();
    }
}

Compilation message (stderr)

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

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