Submission #185756

#	Time	Username	Problem	Language	Result	Execution time	Memory
185756		qkxwsm	JOIRIS (JOI16_joiris)	C++14	30 / 100	3 ms	440 KiB

joiris

#include <bits/stdc++.h>

using namespace std;

template<class T, class U>
void ckmin(T &a, U b)
{
    if (a > b) a = b;
}

template<class T, class U>
void ckmax(T &a, U b)
{
    if (a < b) a = b;
}

#define MP make_pair
#define PB push_back
#define LB lower_bound
#define UB upper_bound
#define fi first
#define se second
#define SZ(x) ((int) (x).size())
#define ALL(x) (x).begin(), (x).end()
#define FOR(i, a, b) for (auto i = (a); i < (b); i++)
#define FORD(i, a, b) for (auto i = (a) - 1; i >= (b); i--)
#define MAXN 51

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef vector<pii> vpi;
typedef vector<pll> vpl;

int N, K;
int arr[MAXN];
int sum[MAXN];
vpi ans;

void solve()
{
    // ////cerr << "PART 1\n";
    // FOR(i, 0, N)
    // {
    //     ////cerr << arr[i] << ' ';
    // }
    // ////cerr << endl;
    FOR(j, 1, N)
    {
        while(arr[j] < arr[j - 1])
        {
            ans.PB({0, j});
            arr[j] += K;
        }
    }
    int sb[K];
    FOR(i, 0, K) sb[i] = 0;
    int d;
    FOR(i, K, N)
    {
        d = arr[i] - arr[i - 1];
        sb[i % K] += d;
        while(d--)
        {
            for (int j = i - K; j >= 0; j -= K)
            {
                ans.PB({1, j});
                FOR(k, j, j + K) arr[k]++;
            }
        }
    }
    //cerr << "PART 1\n";
    FOR(i, 0, N)
    {
        //cerr << arr[i] << ' ';
    }
    //cerr << endl;
    // FOR(i, 0, K) //cerr << sb[i] << endl;
    d = arr[N - 1];
    FOR(i, 0, K)
    {
        FOR(j, 0, N)
        {
            arr[j] -= sb[i];
        }
        //cerr << "PART 2\n";
        FOR(i, 0, N)
        {
            //cerr << arr[i] << ' ';
        }
        //cerr << endl;
        sb[i] = 0;
        // d = arr[K - 1] - arr[i];
        // FOR(j, i + 1, K) d += sb[j];
        if (i == K - 1) break;
        FOR(j, 0, (d + K - 1) / K)
        {
            ans.PB({0, i});
            arr[i] += K;
        }
    }
    FOR(i, 0, N)
    {
        arr[i] -= arr[N - 1];
    }
    // //cerr << "PART 3\n";
    // FOR(i, 0, N)
    // {
    //     //cerr << arr[i] << ' ';
    // }
    // //cerr << endl;
    d = 0;
    FOR(i, N % K, K) ckmax(d, arr[i]);
    FOR(i, N % K, K)
    {
        while(arr[i] < d)
        {
            ans.PB({0, i});
            arr[i] += K;
        }
    }
    FOR(i, 0, N % K) ckmax(d, arr[i]);
    FOR(i, 0, N % K)
    {
        while(arr[i] < d)
        {
            ans.PB({0, i});
            arr[i] += K;
        }
    }
    d = arr[K - 1];
    FOR(i, K, N)
    {
        FOR(j, 0, d / K)
        {
            ans.PB({0, i});
            arr[i] += K;
        }
    }
    FOR(i, 0, N)
    {
        arr[i] -= d;
    }
    // cerr << "PART 4\n";
    // FOR(i, 0, N)
    // {
    //     cerr << arr[i] << ' ';
    // }
    // cerr << endl;
    if (N % K)
    {
        d = arr[0];
        FOR(i, 0, d)
        {
            for (int j = N % K; j < N; j += K)
            {
                ans.PB({1, j});
            }
            FOR(j, 0, N % K)
            {
                arr[j]--;
            }
        }
    }
    // else
    // {
    //     d = arr[0];
    //     FOR(i, 0, N)
    //     {
    //         while(arr[i] < d)
    //         {
    //             ans.PB({0, i});
    //             arr[i] += K;
    //         }
    //     }
    //     FOR(i, 0, N)
    //     {
    //         arr[i] -= d;
    //     }
    // }
    FOR(i, 0, N)
    {
        assert(arr[i] == 0);
    }
    return;
}

int32_t main()
{
    cout << fixed << setprecision(12);
    ////cerr << fixed << setprecision(4);
    ios_base::sync_with_stdio(false); cin.tie(0);
    cin >> N >> K;
    FOR(i, 0, N)
    {
        cin >> arr[i];
        sum[i % K] += arr[i];
    }
    if (K == 1)
    {
        FOR(i, 0, N)
        {
            FOR(j, 0, 50 - arr[i])
            {
                cout << 1 << ' ' << i + 1 << '\n';
            }
        }
        return 0;
    }
    FOR(i, 0, N % K)
    {
        if ((sum[i] - sum[0]) % K)
        {
            cout << "-1\n";
            return 0;
        }
    }
    FOR(i, N % K, K)
    {
        if ((sum[i] - sum[K - 1]) % K)
        {
            cout << "-1\n";
            return 0;
        }
    }
    //it's possible
    solve();
    assert(SZ(ans) <= 10000);
    cout << SZ(ans) << '\n';
    for (pii p : ans)
    {
        cout << p.fi + 1 << ' ' << p.se + 1 << '\n';
    }
    return 0;
}

• Subtask 1: 15 / 15
• Subtask 2: 15 / 15
• Subtask 3: 0 / 15
• Subtask 4: 0 / 55