제출 #961252

#제출 시각아이디문제언어결과실행 시간메모리
961252pragmatistGift (IZhO18_nicegift)C++17
0 / 100
2020 ms476 KiB
#include<bits/stdc++.h>

using namespace std;

const int N = (int)1e6+7;

int n, k, a[N];

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
	cin >> n >> k;
	long long tot = 0;
	int mx = 0;
	for(int i = 1; i <= n; ++i) {
		cin >> a[i];
		tot += a[i];
		mx = max(mx, a[i]);
	}    
	if(tot%k) {
		cout << "-1\n";
		return 0;
	}
	if(mx>tot/2) {
		cout << "-1\n";
		return 0;
	}
	cout << tot/k << "\n";
	set<pair<int, int> > s;
	for(int i = 1; i <= n; ++i) {
		s.insert({a[i], i});
	}
	for(int i = 1; i <= tot/k; ++i) {
		vector<int> cur;
		for(int j = 0; j < k; ++j) {
			auto it = s.begin();
			cur.push_back(it->second);
			s.erase(it);
		}
		cout << "1 ";
		for(auto id : cur) {
			assert(a[id]>0);
			cout << id << ' ';
		}
		cout << "\n";
		for(auto id : cur) {
			a[id]--;
			if(a[id]>0) {
				s.insert({a[id], id});
			}
		}
	}
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...