제출 #334963

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
334963		tengiz05	Nice sequence (IZhO18_sequence)	C++17	100 / 100	1824 ms	51932 KiB

sequence

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define FASTIO ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define all(x) (x).begin(), (x).end()
#define pb push_back
#define pii pair<int, int>
#define ff first
#define ss second
#define PI acos(-1)
#define ld long double
const int mod = 1e9+7, N = 4e6+5;
int msb(int val){return sizeof(int)*8-__builtin_clzll(val);}
int a[N], n, m, k;
bool used[N];
vector<int> order;
int md;
void dfs(int u){
	used[u] = true;
	if(u-n >= 0 && !used[u-n])dfs(u-n);
	if(u+m <= md && !used[u+m])dfs(u+m);
	order.pb(u);
}

bool check(int mid){
	int i;
	md = mid;
	order.clear();
	memset(used, 0, sizeof(used));
	for(i=0;i<=mid;i++)if(!used[i])dfs(i);
	for(i=0;i<=mid;i++)a[order[i]] = i;
	for(i=0;i<=mid;i++){
		if(i-n >= 0){
			if(a[i]-a[i-n] <=0)return false;
		}else if(i-m >= 0){
			if(a[i]-a[i-m] >=0)return false;
		}
	}return true;
}
void solve(int test_case){
	int i, j;
	cin >> n >> m;
	int l = 0, r = n+m+1;
	check(l);
	while(l+1 < r){
		int mid = (l+r)/2;
	//	cout << l << ' ' << r << ' ' << mid << '\n';
		if(check(mid))l = mid;
		else r = mid;
	}
	cout << l << '\n';
	assert(l != -1);
	check(l);
	for(i=1;i<=l;i++){
		cout << -(a[i]-a[i-1]) << ' ';
	}cout << '\n';
	return;
}

signed main(){
	FASTIO;
#define MULTITEST 1
#if MULTITEST
	int ___T;
	cin >> ___T;
	for(int T_CASE = 1; T_CASE <= ___T; T_CASE++)
		solve(T_CASE);
#else
	solve(1);
#endif
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

sequence.cpp: In function 'void solve(long long int)':
sequence.cpp:41:9: warning: unused variable 'j' [-Wunused-variable]
   41 |  int i, j;
      |         ^

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