Submission #347717

#	Time	Username	Problem	Language	Result	Execution time	Memory
347717		Markomafko972	Meteors (POI11_met)	C++14	100 / 100	2600 ms	48316 KiB

met

#include <bits/stdc++.h>
#define X first
#define Y second
#define pb push_back
#define pii pair<int, int>
typedef long long ll;
using namespace std;

const int MOD = 1e9 + 7;
const ll INF = 1e18;
const int OFF = (1 << 20);
const int MAXN = 300005;

int n, m, a, q;
vector<int> v[MAXN];
pair< pii, int > w[MAXN];
ll loga[MAXN];

void update(int x, int y) {
	x++;
	for (; x <= 300000; x += x & -x) loga[x] += y;
}

ll query(int x) {
	x++;
	ll out = 0;
	for (; x > 0; x -= x & -x) out += loga[x];
	return out;
}

vector< pair<pii, int> > gdje[MAXN];
vector< pair<pii, int> > slj[MAXN];
int sol[MAXN];
int treba[MAXN];

int main () {

	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		cin >> a;
		v[a-1].push_back(i);
	}
	
	for (int i = 0; i < n; i++) cin >> treba[i];
	
	cin >> q;
	for (int i = 0; i < q; i++) {
		cin >> w[i].X.X >> w[i].X.Y >> w[i].Y;
		w[i].X.X --;
		w[i].X.Y --;
	}
	
	/*for (int i = 0; i < q; i++) {
		if (w[i].X.X <= w[i].X.Y) {
			update(w[i].X.X, w[i].Y);
			update(w[i].X.Y+1, -w[i].Y);
		}
		else {
			update(w[i].X.X, w[i].Y);	
			update(0, w[i].Y);
			update(w[i].X.Y+1, -w[i].Y);
		}
	}*/
	
	int kol = n;
	for (int i = 0; i < n; i++) {
		gdje[(q-1)/2].push_back({{0, q-1}, i});
	}
	
	memset(sol, -1, sizeof sol);
	int br = 0;
	while (kol > 0) {
		for (int i = 0; i < q; i++) {
			if (w[i].X.X <= w[i].X.Y) {
				update(w[i].X.X, w[i].Y);
				update(w[i].X.Y+1, -w[i].Y);
			}
			else {
				update(w[i].X.X, w[i].Y);
				update(0, w[i].Y);
				update(w[i].X.Y+1, -w[i].Y);
			}
			
			/*cout << "QUERY" << endl;
			for (int j = 0; j < m; j++) cout << query(j) << " ";
			cout << endl; */
			
			for (int j = 0; j < gdje[i].size(); j++) {
				unsigned long long zbr = 0;
				for (int k = 0; k < v[gdje[i][j].Y].size(); k++) {
					//cout << "dodaj poziciju " << v[gdje[i][j].Y][k] << endl;
					zbr += query(v[gdje[i][j].Y][k]);
					if (zbr >= treba[gdje[i][j].Y]) break;
				}
				
				//cout << gdje[i][j].X.X << " do " << gdje[i][j].X.Y << ": " << gdje[i][j].Y << " " << zbr << endl;
				//system("pause");
				
				if (gdje[i][j].X.X == gdje[i][j].X.Y) {
					if (zbr >= treba[gdje[i][j].Y]) sol[gdje[i][j].Y] = i+1;
					else sol[gdje[i][j].Y] = -1;
					
					kol--;
				}
				else {
					if (zbr < treba[gdje[i][j].Y]) slj[(i+1+gdje[i][j].X.Y)/2].push_back({{i+1, gdje[i][j].X.Y}, gdje[i][j].Y});
					else {
						slj[(gdje[i][j].X.X+i)/2].push_back({{gdje[i][j].X.X, i}, gdje[i][j].Y});
					}
				}
			}
		}
		
		for (int i = 0; i < q; i++) {
			gdje[i].clear();
			gdje[i].shrink_to_fit();
			
			gdje[i] = slj[i];
			
			slj[i].clear();
			slj[i].shrink_to_fit();
		}
		
		memset(loga, 0, sizeof loga);
		br++;
	}
	
	for (int i = 0; i < n; i++) {
		if (sol[i] == -1) cout << "NIE\n";
		else cout << sol[i] << "\n";
	}

	return 0;
}

Compilation message (stderr)

met.cpp: In function 'int main()':
met.cpp:88:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<int, int>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   88 |    for (int j = 0; j < gdje[i].size(); j++) {
      |                    ~~^~~~~~~~~~~~~~~~
met.cpp:90:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   90 |     for (int k = 0; k < v[gdje[i][j].Y].size(); k++) {
      |                     ~~^~~~~~~~~~~~~~~~~~~~~~~~
met.cpp:93:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
   93 |      if (zbr >= treba[gdje[i][j].Y]) break;
      |          ~~~~^~~~~~~~~~~~~~~~~~~~~~
met.cpp:100:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
  100 |      if (zbr >= treba[gdje[i][j].Y]) sol[gdje[i][j].Y] = i+1;
      |          ~~~~^~~~~~~~~~~~~~~~~~~~~~
met.cpp:106:14: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
  106 |      if (zbr < treba[gdje[i][j].Y]) slj[(i+1+gdje[i][j].X.Y)/2].push_back({{i+1, gdje[i][j].X.Y}, gdje[i][j].Y});
      |          ~~~~^~~~~~~~~~~~~~~~~~~~~

• 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: 13 / 13
• Subtask 8: 13 / 13