Submission #501591

#	Time	Username	Problem	Language	Result	Execution time	Memory
501591		Nuraly_Serikbay	Meteors (POI11_met)	C++14	0 / 100	6093 ms	58064 KiB

met

#include <bits/stdc++.h>
  
#define endl "\n"
#define mp make_pair
#define pb push_back
#define F first
#define S second
#define sz size()
#define rep(i,k,n) for(int i = k ; i <= n ; ++i)
#define per(i,k,n) for(int i = k ; i >= n ; --i)
#define YOSIK() ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define all(x) x.begin(),x.end()
#define fr(x) freopen(x".in","r",stdin),freopen(x".out","w",stdout)
#define toqta return 0
#define PERMUTE next_permutation
#define no cout<<"No"<<endl;
#define yes cout<<"Yes"<<endl;
#define a() a + 1, a + n + 1  
#define int long long
  
using namespace std;

/*
#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")
*/

typedef long long ll;
typedef unsigned long long ull;
typedef string S;
typedef double D;
typedef long double lld;
   
const int N = 1e6 + 17;
const int modd = 1e9 + 7;
const ll INF = 1e18 - 19;
const int P = 37;
const ll NN = 1e7 + 17;
const D eps = 1e-19;
const double pi = 3.141592653589793238462643383279 ;   
 
struct edge {
	ll a, b, len;	

    bool operator<(const edge& other) {
        return len < other.len;
    }
};


bool sortbysec(const pair<int,int> &a, const pair<int,int> &b){
    return (a.second < b.second);
}
 
void pre (ll a) {
    cout << fixed << setprecision(a);
    return;
}

ll n, m;	
ll a[N], q, sum[N];
vector <ll> v[N];
ll mod[N * 4];
ll L[N], R[N], x[N];

void push (int v, int l, int r) {
	if (mod[v] != 0 && v * 2 + 1 < N * 4) {
		mod[v * 2] += mod[v];
		mod[v * 2 + 1] += mod[v];
		mod[v] = 0;
	}
}

void update (ll l, ll r, ll tl, ll tr, ll v, ll val) {
	if (tr < l || r < tl) return;
	if (l <= tl && tr <= r) {
		mod[v] += val;
		push(v, tl, tr);
		return;
	}
	ll mid = tl + tr >> 1;
	push (v, tl, tr);
	update (l, r, tl, mid, v * 2, val);
	update (l, r, mid + 1, tr, v * 2 + 1, val);
	mod[v] = mod[v * 2] + mod[v * 2 + 1];
	return;
}


ll get (ll v, ll tl, ll tr, ll pos) {
	if (tl == tr) {
		return mod[v];
 	}
 	push(v, tl, tr);
 	int mid = tl + tr >> 1;
 	if (pos <= mid) return get(v * 2, tl, mid, pos);
 	else return get(v * 2 + 1, mid + 1, tr, pos);
}

ll check (ll len, ll pos) {
	ll ans = 0;
	for (int i = 1; i <= len; ++i) {
		if (L[i] > R[i]) {
			update(L[i], m, 1, m, 1, x[i]);				
			update(1, R[i], 1, m, 1, x[i]);	
		} else {
			update(L[i], R[i], 1, m, 1, x[i]);
		}
	}
	for (int i = 0; i < v[pos].size(); ++i) {
		ans += get(1, 1, m, v[pos][i]);
	}
	for (int i = 1; i <= len; ++i) {
		if (L[i] > R[i]) {
			update(L[i], m, 1, m, 1, -x[i]);				
			update(1, R[i], 1, m, 1, -x[i]);	
		} else {
			update(L[i], R[i], 1, m, 1, -x[i]);
		}
	}
	return ans;
}

void Solution (/*Nuraly.Serikbay*/) {
	cin >> n >> m;
	vector <int> west;
	for (int i = 1; i <= m; ++ i) {
		cin >> a[i];
		west.pb(a[i]);
	}
	sort(all(west));
	for (int i = 1; i <= m; ++i) {
		a[i] = lower_bound(all(west), a[i]) - west.begin() + 1;
		v[a[i]].pb(i);
	}
	for (int i = 1; i <= n; ++ i) {
		cin >> sum[i];
	}
	cin >> q;
	for (int i = 1; i <= q; ++ i) {
		cin >> L[i] >> R[i] >> x[i];
	}
	for (int i = 1; i <= n; ++ i) {
		ll l = 0, r = q + 1; 
		while (r - l > 1) {
			ll mid = l + r >> 1;
			if (check (mid, i) >= sum[i]) r = mid;
			else l = mid;
		}
		if (r == q + 1) cout << "NIE\n";
		else cout << r << endl;
	}
	
	return;
}
main () {
	YOSIK();
	ll T = 1;
	//cin >> T;
	while (T --) {
		Solution ();
	}
	exit (0);
}/*

*/

Compilation message (stderr)

met.cpp: In function 'void update(ll, ll, ll, ll, ll, ll)':
met.cpp:89:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   89 |  ll mid = tl + tr >> 1;
      |           ~~~^~~~
met.cpp: In function 'll get(ll, ll, ll, ll)':
met.cpp:103:16: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  103 |   int mid = tl + tr >> 1;
      |             ~~~^~~~
met.cpp: In function 'll check(ll, ll)':
met.cpp:118:20: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  118 |  for (int i = 0; i < v[pos].size(); ++i) {
      |                  ~~^~~~~~~~~~~~~~~
met.cpp: In function 'void Solution()':
met.cpp:154:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  154 |    ll mid = l + r >> 1;
      |             ~~^~~
met.cpp: At global scope:
met.cpp:164:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
  164 | main () {
      | ^~~~

• 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