[Codeforces] Codeforces Round #666 (Div. 2) C. Multiples of Length

2020年12月13日

問題

方針

\( a_i + (n – 1)a_i = na_i \) となることを利用して、\( 1 \leq i \leq n – 1 \) となる \( i\) について、\( (n-1)a_i \) を加算します。その後に、\( -na_i \) を加算することで、\( a_i = 0 \ (1 \leq i \leq n – 1) \) となります。そして、\( a_0 \) に \( -a_0 \) を加算します。

\( n = 1 \) については別途処理します。

コード

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main() {
    int n;
    cin >> n;
    ll a[n];
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    if (n == 1) {
        cout << "1 1\n";
        cout << -a[0] << "\n";
        cout << "1 1\n";
        cout << "0\n";
        cout << "1 1\n";
        cout << "0\n";
        return 0;
    }
    cout << "1 1\n";
    cout << -a[0] << "\n";
    a[0] = 0;
    cout << 2 << " " << n << "\n";
    for (int i = 1; i < n; i++) {
        if (i == n - 1) {
            cout << a[i] * (n - 1)<< "\n";
        } else {
            cout << a[i] * (n - 1) << " ";
        }
        a[i] += a[i] * (n - 1);
    }
    cout << 1 << " " << n << "\n";
    for (int i = 0; i < n; i++) {
        if (i == n - 1) {
            cout << -a[i] << "\n";
        } else {
            cout << -a[i] << " ";
        }
    }
    return 0;
}