#ifndef CPL_TEMPLATE
#define CPL_TEMPLATE
/*
Normie's Template v2.5
Changes:
Added warning against using pragmas on USACO.
*/
// Standard library in one include.
#include <bits/stdc++.h>
using namespace std;
// ordered_set library.
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
#define ordered_set(el) tree<el,null_type,less<el>,rb_tree_tag,tree_order_statistics_node_update>
// AtCoder library. (Comment out these two lines if you're not submitting in AtCoder.) (Or if you want to use it in other judges, run expander.py first.)
//#include <atcoder/all>
//using namespace atcoder;
//Pragmas (Comment out these three lines if you're submitting in szkopul or USACO.)
#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("Ofast,unroll-loops,tree-vectorize")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
//File I/O.
#define FILE_IN "cseq.inp"
#define FILE_OUT "cseq.out"
#define ofile freopen(FILE_IN,"r",stdin);freopen(FILE_OUT,"w",stdout)
//Fast I/O.
#define fio ios::sync_with_stdio(0);cin.tie(0)
#define nfio cin.tie(0)
#define endl "\n"
//Order checking.
#define ord(a,b,c) ((a>=b)and(b>=c))
//min/max redefines, so i dont have to resolve annoying compile errors.
#define min(a,b) (((a)<(b))?(a):(b))
#define max(a,b) (((a)>(b))?(a):(b))
// Fast min/max assigns to use with AVX.
// Requires g++ 9.2.0.
template<typename T>
__attribute__((always_inline)) void chkmin(T& a, const T& b) {
a=(a<b)?a:b;
}
template<typename T>
__attribute__((always_inline)) void chkmax(T& a, const T& b) {
a=(a>b)?a:b;
}
//Constants.
#define MOD (ll(998244353))
#define MAX 300001
#define mag 320
const long double PI=3.14159265358979;
//Pairs and 3-pairs.
#define p1 first
#define p2 second.first
#define p3 second.second
#define fi first
#define se second
#define pii(element_type) pair<element_type,element_type>
#define piii(element_type) pair<element_type,pii(element_type)>
//Quick power of 2.
#define pow2(x) (ll(1)<<x)
//Short for-loops.
#define ff(i,__,___) for(int i=__;i<=___;i++)
#define rr(i,__,___) for(int i=__;i>=___;i--)
//Typedefs.
#define bi BigInt
typedef long long ll;
typedef long double ld;
typedef short sh;
// Binpow and stuff
ll BOW(ll a, ll x, ll p)
{
if (!x) return 1;
ll res=BOW(a,x/2,p);
res*=res;
res%=p;
if (x%2) res*=a;
return res%p;
}
ll INV(ll a, ll p)
{
return BOW(a,p-2,p);
}
//---------END-------//
#endif
vector<ll> vec;
vector<vector<ll>> vec2;
vector<ll> vec3;
ll n,m,i,j,k,t,t1,u,v,a,b;
ll arr[1000001];
ll summ;
struct cmp {
ll operator()(ll i,ll j) const {
if (arr[i]-arr[j]) return (arr[i]>arr[j]);
return (i>j);
}
};
set<ll,cmp> lmao;
int main()
{
fio;
cin>>n>>m;
for (i=1;i<=n;i++) {
cin>>arr[i];
summ+=arr[i];
lmao.insert(i);
}
lmao.insert(0);
while(true) {
auto it = lmao.begin();
vec.clear();
for (i=0;i<m;i++) {
vec.push_back((*it));
it = lmao.erase(it);
}
u=arr[vec[m-1]];
if (arr[(*it)]*m>summ) break;
else u=min(u,(summ-(arr[(*it)]*m))/m);
a=1;
for (auto g : vec) {
if (!arr[g]) a=0;
}
b=(*it);
if (!a) break;
for (auto g : vec) {
arr[g]-=u;
lmao.insert(g);
}
vec2.push_back(vec);
vec3.push_back(u);
summ-=m*u;
}
for (i=1;i<=n;i++) if (arr[i]) {
cout<<"-1";
return 0;
}
cout<<vec2.size()<<endl;
for (i=0;i<vec2.size();i++) {
cout<<vec3[i];
for (auto g : vec2[i]) cout<<' '<<g;
cout<<endl;
}
}
// N;
Compilation message
nicegift.cpp:23: warning: ignoring '#pragma comment ' [-Wunknown-pragmas]
23 | #pragma comment(linker, "/stack:200000000")
|
nicegift.cpp: In function 'int main()':
nicegift.cpp:152:15: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<std::vector<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
152 | for (i=0;i<vec2.size();i++) {
| ~^~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
n=4 |
2 |
Correct |
0 ms |
204 KB |
n=3 |
3 |
Correct |
0 ms |
204 KB |
n=3 |
4 |
Correct |
0 ms |
204 KB |
n=4 |
5 |
Correct |
0 ms |
204 KB |
n=4 |
6 |
Correct |
1 ms |
204 KB |
n=2 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
n=4 |
2 |
Correct |
0 ms |
204 KB |
n=3 |
3 |
Correct |
0 ms |
204 KB |
n=3 |
4 |
Correct |
0 ms |
204 KB |
n=4 |
5 |
Correct |
0 ms |
204 KB |
n=4 |
6 |
Correct |
1 ms |
204 KB |
n=2 |
7 |
Correct |
0 ms |
204 KB |
n=5 |
8 |
Correct |
0 ms |
204 KB |
n=8 |
9 |
Correct |
1 ms |
204 KB |
n=14 |
10 |
Correct |
1 ms |
204 KB |
n=11 |
11 |
Correct |
54 ms |
5404 KB |
n=50000 |
12 |
Correct |
42 ms |
4956 KB |
n=50000 |
13 |
Correct |
1 ms |
204 KB |
n=10 |
14 |
Correct |
1 ms |
332 KB |
n=685 |
15 |
Correct |
1 ms |
332 KB |
n=623 |
16 |
Correct |
1 ms |
332 KB |
n=973 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
n=4 |
2 |
Correct |
0 ms |
204 KB |
n=3 |
3 |
Correct |
0 ms |
204 KB |
n=3 |
4 |
Correct |
0 ms |
204 KB |
n=4 |
5 |
Correct |
0 ms |
204 KB |
n=4 |
6 |
Correct |
1 ms |
204 KB |
n=2 |
7 |
Correct |
0 ms |
204 KB |
n=5 |
8 |
Correct |
0 ms |
204 KB |
n=8 |
9 |
Correct |
1 ms |
204 KB |
n=14 |
10 |
Correct |
1 ms |
204 KB |
n=11 |
11 |
Correct |
54 ms |
5404 KB |
n=50000 |
12 |
Correct |
42 ms |
4956 KB |
n=50000 |
13 |
Correct |
1 ms |
204 KB |
n=10 |
14 |
Correct |
1 ms |
332 KB |
n=685 |
15 |
Correct |
1 ms |
332 KB |
n=623 |
16 |
Correct |
1 ms |
332 KB |
n=973 |
17 |
Correct |
1 ms |
332 KB |
n=989 |
18 |
Correct |
1 ms |
332 KB |
n=563 |
19 |
Correct |
1 ms |
332 KB |
n=592 |
20 |
Correct |
1 ms |
332 KB |
n=938 |
21 |
Correct |
1 ms |
332 KB |
n=747 |
22 |
Correct |
1 ms |
332 KB |
n=991 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1080 ms |
100360 KB |
n=1000000 |
2 |
Correct |
749 ms |
58608 KB |
n=666666 |
3 |
Correct |
405 ms |
32556 KB |
n=400000 |
4 |
Correct |
264 ms |
22388 KB |
n=285714 |
5 |
Correct |
16 ms |
1996 KB |
n=20000 |
6 |
Correct |
196 ms |
16248 KB |
n=181818 |
7 |
Correct |
8 ms |
1100 KB |
n=10000 |
8 |
Correct |
7 ms |
844 KB |
n=6666 |
9 |
Correct |
5 ms |
588 KB |
n=4000 |
10 |
Correct |
7 ms |
716 KB |
n=2857 |
11 |
Correct |
2 ms |
460 KB |
n=2000 |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
204 KB |
n=4 |
2 |
Correct |
0 ms |
204 KB |
n=3 |
3 |
Correct |
0 ms |
204 KB |
n=3 |
4 |
Correct |
0 ms |
204 KB |
n=4 |
5 |
Correct |
0 ms |
204 KB |
n=4 |
6 |
Correct |
1 ms |
204 KB |
n=2 |
7 |
Correct |
0 ms |
204 KB |
n=5 |
8 |
Correct |
0 ms |
204 KB |
n=8 |
9 |
Correct |
1 ms |
204 KB |
n=14 |
10 |
Correct |
1 ms |
204 KB |
n=11 |
11 |
Correct |
54 ms |
5404 KB |
n=50000 |
12 |
Correct |
42 ms |
4956 KB |
n=50000 |
13 |
Correct |
1 ms |
204 KB |
n=10 |
14 |
Correct |
1 ms |
332 KB |
n=685 |
15 |
Correct |
1 ms |
332 KB |
n=623 |
16 |
Correct |
1 ms |
332 KB |
n=973 |
17 |
Correct |
1 ms |
332 KB |
n=989 |
18 |
Correct |
1 ms |
332 KB |
n=563 |
19 |
Correct |
1 ms |
332 KB |
n=592 |
20 |
Correct |
1 ms |
332 KB |
n=938 |
21 |
Correct |
1 ms |
332 KB |
n=747 |
22 |
Correct |
1 ms |
332 KB |
n=991 |
23 |
Correct |
1080 ms |
100360 KB |
n=1000000 |
24 |
Correct |
749 ms |
58608 KB |
n=666666 |
25 |
Correct |
405 ms |
32556 KB |
n=400000 |
26 |
Correct |
264 ms |
22388 KB |
n=285714 |
27 |
Correct |
16 ms |
1996 KB |
n=20000 |
28 |
Correct |
196 ms |
16248 KB |
n=181818 |
29 |
Correct |
8 ms |
1100 KB |
n=10000 |
30 |
Correct |
7 ms |
844 KB |
n=6666 |
31 |
Correct |
5 ms |
588 KB |
n=4000 |
32 |
Correct |
7 ms |
716 KB |
n=2857 |
33 |
Correct |
2 ms |
460 KB |
n=2000 |
34 |
Correct |
37 ms |
3208 KB |
n=23514 |
35 |
Correct |
30 ms |
3204 KB |
n=23514 |
36 |
Correct |
2 ms |
332 KB |
n=940 |
37 |
Correct |
0 ms |
204 KB |
n=2 |
38 |
Correct |
98 ms |
8312 KB |
n=100000 |
39 |
Correct |
104 ms |
8388 KB |
n=100000 |
40 |
Correct |
1 ms |
204 KB |
n=10 |
41 |
Correct |
1 ms |
324 KB |
n=100 |
42 |
Correct |
5 ms |
456 KB |
n=1000 |
43 |
Correct |
1556 ms |
123268 KB |
n=1000000 |
44 |
Execution timed out |
2061 ms |
73956 KB |
Time limit exceeded |
45 |
Halted |
0 ms |
0 KB |
- |