Submission #760568

#	Time	Username	Problem	Language	Result	Execution time	Memory
760568		NK_	Nice sequence (IZhO18_sequence)	C++17	15 / 100	6 ms	980 KiB

sequence

// Success consists of going from failure to failure without loss of enthusiasm
#include <bits/stdc++.h>

using namespace std;

#define nl '\n'
#define pb push_back

using ll = long long;
template<class T> using V = vector<T>;

const int MAX = 1e9;

void solve() {
	int N, M; cin >> N >> M;

	bool SWAP = 0; if (N < M) { swap(N, M); SWAP = 1; }
	// WLOG N > M

	V<int> A;
	if (N % M == 0) {
		int ans = N - 1;
		A = V<int>(ans); for(auto& x : A) x = 1;
	} else {
		int ans = N; A = V<int>(ans);
		int X = -(MAX - 1) / (M - 1);
		int REM = (MAX - 1) % (M - 1);
		// cout << REM << nl;
		// cout << X << nl;

		for(int i = 0; i < ans; i++) {
			if (i % M == M - 1) A[i] = MAX;
			else {
				// cout << i % M << " " << REM << endl;
				A[i] = ((i % M) < REM ? X - 1 : X);
			}
		}
	}

	// cout << SWAP << nl;
	if (SWAP) for(auto& x : A) x = -x;

	cout << size(A) << nl;
	for(auto& x : A) cout << x << " ";
	cout << nl;
	
	ll SUM = accumulate(begin(A), end(A), 0LL);
	// cout << SUM << nl;

}	

int main() {
	cin.tie(0)->sync_with_stdio(0);

	int T; cin >> T;
	while(T--) {
		solve();
	}

    return 0;
}

Compilation message (stderr)

sequence.cpp: In function 'void solve()':
sequence.cpp:47:5: warning: unused variable 'SUM' [-Wunused-variable]
   47 |  ll SUM = accumulate(begin(A), end(A), 0LL);
      |     ^~~

• Subtask 1: 6 / 6
• Subtask 2: 9 / 9
• Subtask 3: 0 / 14
• Subtask 4: 0 / 15
• Subtask 5: 0 / 14
• Subtask 6: 0 / 18
• Subtask 7: 0 / 24