Submission #667872

#	Time	Username	Problem	Language	Result	Execution time	Memory
667872		Kaztaev_Alisher	Gift (IZhO18_nicegift)	C++17	30 / 100	293 ms	46072 KiB

nicegift

//#pragma GCC optomize ("Ofast")
//#pragma GCC optomize ("unroll-loops")
//#pragma GCC target ("avx,avx2,fma")
 
#include <bits/stdc++.h>
#define F first
#define S second
#define pb push_back
#define sz size
#define cl clear
#define ins insert
#define ers erase
#define pii pair < int , int >
#define pll pair< long long  , long long >
#define all(x) x.begin() , x.end()
#define ios ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define tostr(x) to_string(x)
#define tonum(s) atoi(s.c_str())
#define seon(x) setprecision(x)
#define bpop(x) __builtin_popcount(x)
#define deb(x) cerr << #x  << " = " << x << endl;
 
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef long double ldb;
const double PI = 3.14159265;
 
const ll N = 1e6+5;
const ll mod = 1e9+7;
const ll inf = 1e9;
const ll INF = 1e18;
 
using namespace std;
 
set<pll>st;
pll b[N];
ll a[N];
vector<int> v[N];
void solve(){
	ll n , k ,cnt = 0 , kal = 0;
	cin >> n >> k;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		if(a[i] == a[i-1]) kal++;
	}
	if(kal == n-1){
		if(n % k != 0){
			cout << "-1\n";
		} else {
			kal = 1;
			cout << n/k <<"\n";
			for(int j = 1; j <= n/k; j++){
				cout << a[1] <<" ";
				for(int i = kal; i < kal+k; i++){
					cout << i <<" ";	
				}
				kal += k;
				cout << "\n";
			}
		}
		return;
	}
	for(int i = 1; i <= n; i++){
		st.ins({a[i] , i});
	}
	while(st.sz()){
		++cnt;
		if(st.sz() < k){
			cout << "-1\n";
			return;
		}
		for(int i = 1; i <= k; i++){
			b[i] = *st.rbegin();
			st.ers(*st.rbegin());
			v[cnt].pb(b[i].S);
		}
		for(int i = 1; i <= k; i++) {
			b[i].F--;
			if(b[i].F) st.ins({b[i].F , b[i].S});
		}
	}
	cout << cnt <<"\n";
	for(int i = 1; i <= cnt; i++){
		cout << 1 <<" ";
		for(int x : v[i]) cout << x <<" ";
		cout << "\n";
	}
}
signed main(){
	ios;
	solve();
	return 0;
}
/*
*/

Compilation message (stderr)

nicegift.cpp: In function 'void solve()':
nicegift.cpp:70:14: warning: comparison of integer expressions of different signedness: 'std::set<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} and 'll' {aka 'long long int'} [-Wsign-compare]
   70 |   if(st.sz() < k){
      |      ~~~~~~~~^~~

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