oj.uz

Submission #90937

# Submission time Handle Problem Language Result Execution time Memory
90937 2018-12-25T09:03:58 Z toloraia Gift (IZhO18_nicegift) C++14
30 / 100
 1050 ms 147780 KB 
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define F first
#define S second

using namespace std;

const ll N = 5e6 + 5, INF = 1e9 + 5;

ll n, k;
pair < ll, ll > P[N];
ll nn;
vector < ll > V[N], v;
ll S;

main()
{
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for (ll i = 1; i <= n; i++){
        cin>>P[i].F;
        P[i].S = i;
        S += P[i].F;
    }
    sort (P + 1, P + n + 1);
    reverse (P + 1, P + n + 1);
    if (S <= 1e5){
        while (P[k].F > 0){
            ll I1 = k, I2 = k;
            for (ll i = 20; i >= 0; i--){
                if (I1 - (1<<i) <= 0)
                    continue;
                if (P[I1 - (1<<i)].F == P[k].F)
                    I1 -= (1<<i);
            }
            for (ll i = 20; i >= 0; i--){
                if (I2 + (1<<i) > n)
                    continue;
                if (P[I2 + (1<<i)].F == P[k].F)
                    I2 += (1<<i);
            }
            v.clear();
            for (ll i = 1; i < I1; i++){
                v.pb (P[i].S);
                P[i].F--;
            }
            for (ll i = I2; (ll)v.size() < k; i--){
                v.pb (P[i].S);
                P[i].F--;
            }
            V[++nn] = v;
        }
        if (P[1].F > 0){
            cout<<-1<<endl;
            return 0;
        }
        cout<<nn<<endl;
        for (ll i = 1; i <= nn; i++){
            cout<<"1 ";
            for (ll j = 0; j < k; j++)
                cout<<V[i][j]<<" ";
            cout<<endl;
        }
        return 0;
    }
    if (P[1].F == P[n].F){
        ll d = __gcd (n, k);
        ll m = k / d;
        if (P[1].F % m != 0){
            cout<<-1<<endl;
            return 0;
        }
        ll p = P[1].F / m;
        if (n * m > 3e6){
            assert(0);
            return 0;
        }
        cout<<n / d<<endl;
        for (ll i = 1; i <= n; i += d){
            cout<<p<<" ";
            for (ll j = i; j < i + d; j++)
                cout<<(j - 1) % n + 1<<" ";
            cout<<endl;
        }
        return 0;
    }
    return 0;
}

Compilation message

nicegift.cpp:17:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main()
      ^

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 90 ms 117752 KB n=4
2 Correct 90 ms 117920 KB n=3
3 Correct 96 ms 117920 KB n=3
4 Correct 107 ms 117920 KB n=4
5 Correct 104 ms 117920 KB n=4
6 Correct 107 ms 117968 KB n=2

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 90 ms 117752 KB n=4
2 Correct 90 ms 117920 KB n=3
3 Correct 96 ms 117920 KB n=3
4 Correct 107 ms 117920 KB n=4
5 Correct 104 ms 117920 KB n=4
6 Correct 107 ms 117968 KB n=2
7 Correct 108 ms 117968 KB n=5
8 Correct 108 ms 118356 KB n=8
9 Correct 127 ms 118512 KB n=14
10 Correct 118 ms 118512 KB n=11
11 Correct 194 ms 121100 KB n=50000
12 Correct 195 ms 121100 KB n=50000
13 Correct 184 ms 121100 KB n=10
14 Correct 153 ms 121100 KB n=685
15 Correct 275 ms 121100 KB n=623
16 Correct 130 ms 121100 KB n=973

Subtask #3
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 90 ms 117752 KB n=4
2 Correct 90 ms 117920 KB n=3
3 Correct 96 ms 117920 KB n=3
4 Correct 107 ms 117920 KB n=4
5 Correct 104 ms 117920 KB n=4
6 Correct 107 ms 117968 KB n=2
7 Correct 108 ms 117968 KB n=5
8 Correct 108 ms 118356 KB n=8
9 Correct 127 ms 118512 KB n=14
10 Correct 118 ms 118512 KB n=11
11 Correct 194 ms 121100 KB n=50000
12 Correct 195 ms 121100 KB n=50000
13 Correct 184 ms 121100 KB n=10
14 Correct 153 ms 121100 KB n=685
15 Correct 275 ms 121100 KB n=623
16 Correct 130 ms 121100 KB n=973
17 Correct 143 ms 121100 KB n=989
18 Correct 117 ms 121100 KB n=563
19 Correct 109 ms 121100 KB n=592
20 Correct 114 ms 121100 KB n=938
21 Correct 124 ms 121100 KB n=747
22 Correct 122 ms 121100 KB n=991

Subtask #4
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 1050 ms 147780 KB n=1000000
2 Correct 595 ms 147780 KB n=666666
3 Correct 320 ms 147780 KB n=400000
4 Incorrect 566 ms 147780 KB Expected int32, but "500000500000" found
5 Halted 0 ms 0 KB -

Subtask #5
0 / 51.0

# Verdict Execution time Memory Grader output
1 Correct 90 ms 117752 KB n=4
2 Correct 90 ms 117920 KB n=3
3 Correct 96 ms 117920 KB n=3
4 Correct 107 ms 117920 KB n=4
5 Correct 104 ms 117920 KB n=4
6 Correct 107 ms 117968 KB n=2
7 Correct 108 ms 117968 KB n=5
8 Correct 108 ms 118356 KB n=8
9 Correct 127 ms 118512 KB n=14
10 Correct 118 ms 118512 KB n=11
11 Correct 194 ms 121100 KB n=50000
12 Correct 195 ms 121100 KB n=50000
13 Correct 184 ms 121100 KB n=10
14 Correct 153 ms 121100 KB n=685
15 Correct 275 ms 121100 KB n=623
16 Correct 130 ms 121100 KB n=973
17 Correct 143 ms 121100 KB n=989
18 Correct 117 ms 121100 KB n=563
19 Correct 109 ms 121100 KB n=592
20 Correct 114 ms 121100 KB n=938
21 Correct 124 ms 121100 KB n=747
22 Correct 122 ms 121100 KB n=991
23 Correct 1050 ms 147780 KB n=1000000
24 Correct 595 ms 147780 KB n=666666
25 Correct 320 ms 147780 KB n=400000
26 Incorrect 566 ms 147780 KB Expected int32, but "500000500000" found
27 Halted 0 ms 0 KB -