Submission #667283

#	Time	Username	Problem	Language	Result	Execution time	Memory
667283		Kalashnikov	Nice sequence (IZhO18_sequence)	C++17	15 / 100	9 ms	1076 KiB

sequence

#include <bits/stdc++.h>
 
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
 
using namespace std;
using ll = long long;
 
const int N = 2e5+5 , inf = 2e9 + 7;
const ll INF = 1e18 ,   mod = 1e9+7 , P = 6547;

void solve(int tc) {
	int n, m;
	cin >> n >> m;
	int coef = 1;
	if(n > m) {
		swap(n , m);
		coef *= -1;
	}
	if(m % n == 0) {
		cout << m-1 << '\n';
		for(int i = 1; i < m; i ++) {
			cout << (-1)*coef << ' ';
		}	
		cout << '\n';
		// return;c
		return;
	}
	int mx = 9e8;
	int otr = -mx/(n-1);
	int psv = -otr*(n-1)-1;
	vector<int> vec;
	int sum = 0;
	for(int i = 1; i <= m; i ++) {
		if(i%n == m%n) {
			vec.push_back(psv);
		}
		else {
			vec.push_back(otr);
		}
		sum += vec.back();
	}
	int cur = 0;
	for(int i = m+1; sum > 0; i ++) {
		if(i%n == m%n) {
			vec.push_back(psv);
		}
		else {
			vec.push_back(otr);
		}
		sum += vec.back();
		sum -= vec[cur++];
		if(sum <= 0) {
			vec.pop_back();
			break;
		}
	}	
	psv = mx/(n-1);
	otr = -psv*(n-1)-1;
	vector<int> vec1;
	sum = 0;
	for(int i = 1; i <= m; i ++) {
		if(i%n == (m-1)%n) {
			vec1.push_back(otr);
		}
		else {
			vec1.push_back(psv);
		}
		sum += vec1.back();
	}
	cur = 0;
	for(int i = m+1; sum > 0; i ++) {
		if(i%n == (m-1)%n) {
			vec1.push_back(otr);
		}
		else {
			vec1.push_back(psv);
		}
		sum += vec1.back();
		sum -= vec1[cur++];
		if(sum <= 0) {
			vec1.pop_back();
			break;
		}
	}	
	if(1) swap(vec,vec1);
	cout << vec.size() << '\n';
	for(auto to: vec) {
		cout << to*coef << ' ';
	}
	cout << '\n';
}
/*
*/
main() {
    ios;
    int tt = 1 , tc = 0;
    cin >> tt;
    while(tt --) {
        solve(++tc);
    }
    return 0;
}

Compilation message (stderr)

sequence.cpp:98:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   98 | main() {
      | ^~~~

• 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