Submission #90466

#	Time	Username	Problem	Language	Result	Execution time	Memory
90466		inom	Gift (IZhO18_nicegift)	C++14	30 / 100	53 ms	14268 KiB

nicegift

#include<bits/stdc++.h>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>

#define fi first
#define se second
#define new new228
#define pb push_back
#define rank rank228
#define sz(c) (int)(c).size()
#define all(c) (c).begin(), (c).end()
#define rall(c) (c).rbegin(), (c).rend()
 
using namespace std;
using namespace __gnu_pbds;
 
#pragma GCC optimize("Ofast")
#pragma GCC optimize("no-stack-protector")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,popcnt,abm,mmx,tune=native")
#pragma GCC optimize("fast-math")
#pragma warning(disable : 4996)
 
typedef tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set; // st.oreder_of_key();

const int N = 100100;
const int MAXN = 4 * N;
const int INF = 1e9 + 7;
const int MOD = 998244353;
 
int TN = 1;

int n, k;
int a[N], cnt;
vector<int> ans[N];
set<pair<int, int>> st;

void solve() {
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
		st.insert({a[i], i});
	}
	while (!st.empty()) {
		if (sz(st) < k) {
			printf("-1\n"); return;
		}
		cnt++;
		int cur = k;
		vector<pair<int, int>> tmp;
		while (cur--) {
			auto now = *st.rbegin();
			st.erase(--st.end());
			ans[cnt].pb(now.se);
			now.fi--;
			tmp.pb(now);
		}
		for (auto i: tmp) {
			if (i.fi) {
				st.insert(i);
			}
		}
	}
	printf("%d\n", cnt);
	for (int i = 1; i <= cnt; i++) {
		printf("%d ", 1);
		for (auto j: ans[i]) {
			printf("%d ", j);
		}
		puts("");
	}
	return;
}

signed main() {
    // in; out;  // cin >> TN;
    while (TN--) { solve(); }
    return 0;
}

Compilation message (stderr)

nicegift.cpp:22:0: warning: ignoring #pragma warning  [-Wunknown-pragmas]
 #pragma warning(disable : 4996)
 
nicegift.cpp: In function 'void solve()':
nicegift.cpp:39:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d %d", &n, &k);
  ~~~~~^~~~~~~~~~~~~~~~~
nicegift.cpp:41:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d", &a[i]);
   ~~~~~^~~~~~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 12 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 51