#693008 (C++17) No.1658 Product / Sum

• 問題: No.1658 Product / Sum
• ユーザー: 沙耶花
• 提出日時: 2021-08-27 21:54:13
• 言語: C++17 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 13 ms / 2,000 ms
• コード長: 1,146 bytes
• コンパイル時間: 4,261 ms
• コンパイル使用メモリ: 261,184 KB
• 実行使用メモリ: 4,380 KB
• 最終ジャッジ日時: 2023-08-13 08:45:27
• 合計ジャッジ時間: 6,807 ms
• ジャッジサーバーID （参考情報）: judge15 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,380 KB
testcase_01	AC	2 ms; 4,380 KB
testcase_02	AC	1 ms; 4,376 KB
testcase_03	AC	2 ms; 4,376 KB
testcase_04	AC	5 ms; 4,380 KB
testcase_05	AC	8 ms; 4,376 KB
testcase_06	AC	13 ms; 4,376 KB
testcase_07	AC	5 ms; 4,376 KB
testcase_08	AC	9 ms; 4,380 KB
testcase_09	AC	7 ms; 4,380 KB
testcase_10	AC	4 ms; 4,380 KB
testcase_11	AC	3 ms; 4,380 KB
testcase_12	AC	7 ms; 4,376 KB
testcase_13	AC	6 ms; 4,376 KB
testcase_14	AC	11 ms; 4,376 KB
testcase_15	AC	4 ms; 4,376 KB

ソースコード

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000001
pair<long long,long long> NG(-1,-1);

long long get(long long n){
	long long ok = 0,ng = 1000000005;
	while(ng-ok>1){
		long long mid = (ok+ng)/2;
		if(mid*mid<=n)ok = mid;
		else ng = mid;
	}
	if(ok*ok!=n)return -1;
	return ok;
}

pair<long long,long long> get(long long a,long long b){
	
	long long t = a*a - b*4;
	if(t<0)return NG;
	t = get(t);
	if(t<0)return NG;
	pair<long long,long long> ret;
	ret.first = a+t;
	ret.second = a-t;
	if(ret.first%2!=0)return NG;
	ret.first /= 2;
	ret.second /= 2;
	//if(a==36)cout<<ret.first<<','<<ret.second<<endl;
	if(min(ret.first,ret.second)<=0)return NG;
	return ret;
	
	
}

int main(){
	
	long long N,K;
	cin>>N>>K;
	
	vector<long long> ans(N,1);
	
	for(int i=N;true;i++){
		int S = i;
		pair<long long,long long> p = get(S-(N-2),K*S);
		if(p==NG)continue;
		ans[0] = p.first;
		ans[1] = p.second;
		break;
	}
	
	rep(i,N){
		if(i!=0)cout<<' ';
		cout<<ans[i];
	}
	cout<<endl;
	
	return 0;
}