Submission #342160

#	Time	Username	Problem	Language	Result	Execution time	Memory
342160		maskoff	Nice sequence (IZhO18_sequence)	C++17	15 / 100	6 ms	876 KiB

sequence

#include <bits/stdc++.h>

#define file ""

#define all(x) x.begin(), x.end()

#define sc second
#define fr first

#define pb push_back
#define mp make_pair

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;

const ll inf = 1e18 + 5;
const ll mod = 1e9 + 7;

const int N = 1e6 + 5;

int dx[] = {+1, 0, -1, 0};
int dy[] = {0, +1, 0, -1};

int n, m;

void solve1() {
 	cout << max(n, m) - 1 << endl;
 	for (int i = 1; i < max(n, m); i++) 
 		cout << (min(n, m) == n ? -1 : 1) << " ";
	cout << endl;
}

void solve2() {
	cout << max(n, m) << endl;
	cout << (n == 2 ? 1 : -1) * (max(n, m) / 2 + 1) << " ";
 	for (int i = 1; i <= max(n, m) / 2; i++) 
 		if (n == 2) cout << "-1000000 999999 ";
 		else cout << "1000000 -999999 ";
	cout << endl;
}

vector<int> a;
bool solve3() {
	if (a.size() == max(n, m)) {
	 	for (int i = 0; i < a.size(); i++) {
	 		ll sum = 0;
	 		if (i + n - 1 >= a.size()) continue;
	 		int cnt = n;
	 		for (int j = i; cnt > 0; j++, cnt--) sum += a[j];
	 		if (sum >= 0) return 0;
	 	}
	 	for (int i = 0; i < a.size(); i++) {
	 		ll sum = 0;
	 		if (i + m - 1 >= a.size()) continue;
	 		int cnt = m;
	 		for (int j = i; cnt > 0; j++, cnt--) sum += a[j];
	 		if (sum <= 0) return 0;
	 	}
	 	for (int to : a) cout << to << " ";
	 	cout << endl;
		return 1;
	}
	for (int i = -3; i <= 3; i++) {
		if (i == 0) continue;
	 	a.pb(i);
	 	if (solve3()) return 1;
	 	a.pop_back();
	}
	return 0;
}

void solve4() {
 	int k = min(n, m);
 	vector<int> ans;
 	for (int i = 1; i < (k + 1) / 2; i++)
 		ans.pb(-1e6);
 	if (n % 2 == 0) ans.pb(1);
 	else ans.pb(-1e6 + 1);
 	for (int i = 1; i < (k + 1) / 2; i++)
 		ans.pb(1e6);
 	vector<int> all;
 	int last = 0;
 	while (all.size() < max(n, m)) {
 	 	all.pb(ans[last]);
 	 	last++;
 	 	if (last == k) last = 0;
 	}
 	cout << max(n, m) << endl;
 	for (int to : all) cout << to * (n == min(n, m) ? +1 : -1) << " ";
 	cout << endl;
}

void solve() {
 	cin >> n >> m;
 	a.clear();
 	if (max(n, m) % min(n, m) == 0) {solve1(); return;}
 	if (min(n, m) == 2) {solve2(); return;}
 	solve4();
}

int main() { 
	ios_base :: sync_with_stdio(false);               
	cin.tie(nullptr);
	srand(time(nullptr));
	int t;
	cin >> t;
	while (t--) solve();
  return 0;
}

Compilation message (stderr)

sequence.cpp: In function 'bool solve3()':
sequence.cpp:46:15: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'const int' [-Wsign-compare]
   46 |  if (a.size() == max(n, m)) {
      |      ~~~~~~~~~^~~~~~~~~~~~
sequence.cpp:47:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   47 |    for (int i = 0; i < a.size(); i++) {
      |                    ~~^~~~~~~~~~
sequence.cpp:49:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   49 |     if (i + n - 1 >= a.size()) continue;
      |         ~~~~~~~~~~^~~~~~~~~~~
sequence.cpp:54:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   54 |    for (int i = 0; i < a.size(); i++) {
      |                    ~~^~~~~~~~~~
sequence.cpp:56:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |     if (i + m - 1 >= a.size()) continue;
      |         ~~~~~~~~~~^~~~~~~~~~~
sequence.cpp: In function 'void solve4()':
sequence.cpp:85:21: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'const int' [-Wsign-compare]
   85 |   while (all.size() < max(n, m)) {
      |          ~~~~~~~~~~~^~~~~~~~~~~

• 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