Submission #660913

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
660913	2022-11-23T14:18:05 Z	rukashii	Meteors (POI11_met)	C++17	74 / 100	955 ms	65536 KB

#include <bits/stdc++.h>
using namespace std;
 
#define int long long
 
const int maxn = 3e5 + 2;
 
int n, m, a[maxn], q, tree[maxn], ans[maxn], bad[maxn];
vector <tuple<int, int, int>> query;
vector <int> st[maxn], check[maxn];
pair <int, int> b[maxn];
 
void update(int pos, int val)
{
	for (int i = pos; i <= m; i += (i & (-i)))
	{
		tree[i] += val;
	}
}
 
int get(int pos)
{
	int ret = 0;
	for (int i = pos; i >= 1; i -= (i & (-i)))
	{
		ret += tree[i];
	}
	
	return ret;
}
 
void query_apply(int i)
{
	auto [u, v, f] = query[i - 1];
	
	if (u <= v)
	{
		update(u, f);
		update(v + 1, -f);
	}
	else
	{
		update(1, f);
		update(v + 1, -f);
		update(u, f);
	}
}
 
signed main()
{
	ios::sync_with_stdio(0); cin.tie(0);
	
	memset(ans, -1, sizeof(ans));
	memset(bad, -1, sizeof(bad));
 
	//freopen("input.inp", "r", stdin);
	//freopen("output.out", "w", stdout);
	
	cin >> n >> m;
	
	for (int i = 1, t; i <= m; i++)
	{
		cin >> t;
		
		st[t].push_back(i);
	}
	
	for (int i = 1; i <= n; i++)
		cin >> a[i];
		
	cin >> q;
	
	for (int i = 1; i <= q; i++)
	{
		int l, r, w;
		cin >> l >> r >> w;
		
		query.push_back({l, r, w});
	}
	
	for (int i = 1; i <= n; i++)
	{
	    b[i] = {1, q + 1};
	}
	
	int cnt = n;
	while (1)
	{
		bool kt = 0;
		memset(tree, 0, sizeof(tree));
		for (int i = 1; i <= q; i++)
			check[i].clear();
		
		for (int i = 1; i <= n; i++)
		{
			auto [l, r] = b[i];
			if (l < r)
			{
				int mid = (l + r) / 2;
				check[mid].push_back(i);
				kt = 1;
			}
		}
		
		if (!kt)
			break;
		
		for (int i = 1; i <= q; i++)
		{
			query_apply(i);
			
			for (int p : check[i])
			{
				int &l = b[p].first, &r = b[p].second;
				int mid = (l + r) / 2;
				
				int sum = 0;
				for (int child : st[p])
				{
					sum += get(child);
					if (sum >= a[p])
						break;
				}
				
				
				if (sum < a[p])
				{
					l = mid + 1;
				}
				else
				{
					r = mid;
				}
			}
		}
	}
	
	
 
	for (int i = 1; i <= n; i++)
        if (b[i].second <= q)
        	cout << b[i].second << '\n';
        else
        	cout << "NIE\n";
}

Compilation message

met.cpp: In function 'int main()':
met.cpp:86:6: warning: unused variable 'cnt' [-Wunused-variable]
   86 |  int cnt = n;
      |      ^~~

Subtask #1
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	12 ms	21460 KB	Output is correct
2	Correct	13 ms	21460 KB	Output is correct
3	Correct	11 ms	21472 KB	Output is correct

Subtask #2
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	11 ms	21460 KB	Output is correct
2	Correct	11 ms	21576 KB	Output is correct
3	Correct	14 ms	21608 KB	Output is correct

Subtask #3
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	61 ms	24128 KB	Output is correct
2	Correct	107 ms	27568 KB	Output is correct
3	Correct	92 ms	26264 KB	Output is correct

Subtask #4
12.0 / 12.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	90 ms	25444 KB	Output is correct
2	Correct	110 ms	25552 KB	Output is correct
3	Correct	105 ms	27676 KB	Output is correct
4	Correct	34 ms	24472 KB	Output is correct

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	77 ms	24532 KB	Output is correct
2	Correct	90 ms	27804 KB	Output is correct
3	Correct	50 ms	22728 KB	Output is correct
4	Correct	97 ms	26872 KB	Output is correct

Subtask #6
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	81 ms	23280 KB	Output is correct
2	Correct	97 ms	25384 KB	Output is correct
3	Correct	73 ms	23740 KB	Output is correct
4	Correct	113 ms	28992 KB	Output is correct

Subtask #7
0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	953 ms	52880 KB	Output is correct
2	Correct	611 ms	34512 KB	Output is correct
3	Correct	222 ms	29972 KB	Output is correct
4	Runtime error	955 ms	65536 KB	Execution killed with signal 9
5	Halted	0 ms	0 KB	-

Subtask #8
0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	935 ms	50640 KB	Output is correct
2	Correct	657 ms	34512 KB	Output is correct
3	Correct	183 ms	29612 KB	Output is correct
4	Runtime error	938 ms	65536 KB	Execution killed with signal 9
5	Halted	0 ms	0 KB	-