oj.uz

Submission #92169

# Submission time Handle Problem Language Result Execution time Memory
92169 2019-01-01T19:44:43 Z davitmarg Gift (IZhO18_nicegift) C++17
30 / 100
 2000 ms 89756 KB 
/*
DEATH-MATCH
Davit-Marg
*/
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <string>
#include <cstring>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <stack>
#include <iterator>
#include <ctype.h>
#include <stdlib.h>  
#include <cassert>
#include <fstream>  
#define mod 998244353ll
#define LL long long
#define LD long double
#define MP make_pair
#define PB push_back
using namespace std;

LL gcd(LL a, LL b) {
	while (b) {
		a %= b;
		swap(a, b);
	}
	return a;
}

LL n,k,a[1000006],d,v,s;
bool hav=1;
priority_queue<pair<LL, int>> q;
vector<int> ans;	
int main()
{
	cin >> n >> k;
	for (LL i = 0; i < n; i++)
	{
		scanf("%lld", a + i);
		s += a[i];
		q.push(MP(a[i],i));
		if (i && a[i] != a[i - 1])
			hav = 0;
	}


	if (hav && 0)
	{
		d = k / gcd(n, k);
		if (a[0] % d != 0)
		{
			cout << -1 << endl;
			return 0;
		}
		cout << d * n / k;
		for (LL i = 0; i < d*n; i++)
		{
			v %= k;
			if (v == 0)
				cout << endl << a[0] / d << " ";
			cout << (i%n) + 1 << " ";
			v++;
		}
		cout << endl;
	}
	else
	{
		int step = 0;
		vector<pair<LL, int>> st;
		while (!q.empty())
		{
			while (step == 0 && !st.empty())
			{
				q.push(st[st.size()-1]);
				st.pop_back();
			}
			pair<LL, int> p=q.top();
			q.pop();
			ans.PB(p.second);
			p.first--;
			if (p.first > 0)
				st.PB(p);
			step++;
			step %= k;
		}
		if (!q.empty() || !st.empty())
		{
			cout << -1 << endl;
			return 0;
		}
		else
		{
			cout << ans.size() / k;
			for (int i = 0; i < ans.size(); i++)
			{
				if (i%k == 0)
					cout<< endl << 1 << " ";
				cout << ans[i]+1 << " ";
			}
			cout << endl;
		}
	}
	return 0;
}

/*

4 2
1 1 4 4 

4 3
6 6 6 6 

14 7
10 10 10 10 10 10 10 10 10 10 10 10 10 10

*/

Compilation message

nicegift.cpp: In function 'int main()':
nicegift.cpp:100:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for (int i = 0; i < ans.size(); i++)
                    ~~^~~~~~~~~~~~
nicegift.cpp:45:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lld", a + i);
   ~~~~~^~~~~~~~~~~~~~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB n=4
2 Correct 2 ms 256 KB n=3
3 Correct 2 ms 256 KB n=3
4 Correct 2 ms 252 KB n=4
5 Correct 2 ms 256 KB n=4
6 Correct 2 ms 376 KB n=2

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB n=4
2 Correct 2 ms 256 KB n=3
3 Correct 2 ms 256 KB n=3
4 Correct 2 ms 252 KB n=4
5 Correct 2 ms 256 KB n=4
6 Correct 2 ms 376 KB n=2
7 Correct 2 ms 376 KB n=5
8 Correct 3 ms 504 KB n=8
9 Correct 25 ms 676 KB n=14
10 Correct 16 ms 632 KB n=11
11 Correct 99 ms 3304 KB n=50000
12 Correct 94 ms 3308 KB n=50000
13 Correct 83 ms 1212 KB n=10
14 Correct 74 ms 1264 KB n=685
15 Correct 85 ms 1260 KB n=623
16 Correct 46 ms 1012 KB n=973

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB n=4
2 Correct 2 ms 256 KB n=3
3 Correct 2 ms 256 KB n=3
4 Correct 2 ms 252 KB n=4
5 Correct 2 ms 256 KB n=4
6 Correct 2 ms 376 KB n=2
7 Correct 2 ms 376 KB n=5
8 Correct 3 ms 504 KB n=8
9 Correct 25 ms 676 KB n=14
10 Correct 16 ms 632 KB n=11
11 Correct 99 ms 3304 KB n=50000
12 Correct 94 ms 3308 KB n=50000
13 Correct 83 ms 1212 KB n=10
14 Correct 74 ms 1264 KB n=685
15 Correct 85 ms 1260 KB n=623
16 Correct 46 ms 1012 KB n=973
17 Correct 66 ms 1520 KB n=989
18 Correct 21 ms 884 KB n=563
19 Correct 26 ms 1264 KB n=592
20 Correct 29 ms 1260 KB n=938
21 Correct 25 ms 1012 KB n=747
22 Correct 26 ms 1136 KB n=991

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Execution timed out 2047 ms 89756 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #5
0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB n=4
2 Correct 2 ms 256 KB n=3
3 Correct 2 ms 256 KB n=3
4 Correct 2 ms 252 KB n=4
5 Correct 2 ms 256 KB n=4
6 Correct 2 ms 376 KB n=2
7 Correct 2 ms 376 KB n=5
8 Correct 3 ms 504 KB n=8
9 Correct 25 ms 676 KB n=14
10 Correct 16 ms 632 KB n=11
11 Correct 99 ms 3304 KB n=50000
12 Correct 94 ms 3308 KB n=50000
13 Correct 83 ms 1212 KB n=10
14 Correct 74 ms 1264 KB n=685
15 Correct 85 ms 1260 KB n=623
16 Correct 46 ms 1012 KB n=973
17 Correct 66 ms 1520 KB n=989
18 Correct 21 ms 884 KB n=563
19 Correct 26 ms 1264 KB n=592
20 Correct 29 ms 1260 KB n=938
21 Correct 25 ms 1012 KB n=747
22 Correct 26 ms 1136 KB n=991
23 Execution timed out 2047 ms 89756 KB Time limit exceeded
24 Halted 0 ms 0 KB -