#331206 (C++14) No.550 夏休みの思い出（１）

提出ソース

結果

問題	No.550 夏休みの思い出（１）
ユーザー	minami
提出日時	2019-04-01 18:09:49
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2 ms / 2,000 ms
コード長	2,765 bytes
コンパイル時間	1,786 ms
コンパイル使用メモリ	180,496 KB
実行使用メモリ	5,376 KB
最終ジャッジ日時	2024-05-04 04:31:05
合計ジャッジ時間	3,290 ms
ジャッジサーバーID （参考情報）	judge5 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	1 ms 5,376 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	1 ms 5,376 KB
testcase_11	AC	2 ms 5,376 KB
testcase_12	AC	2 ms 5,376 KB
testcase_13	AC	2 ms 5,376 KB
testcase_14	AC	2 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	1 ms 5,376 KB
testcase_20	AC	2 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	1 ms 5,376 KB
testcase_24	AC	1 ms 5,376 KB
testcase_25	AC	1 ms 5,376 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	2 ms 5,376 KB
testcase_29	AC	1 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	2 ms 5,376 KB
testcase_32	AC	1 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	1 ms 5,376 KB
testcase_35	AC	1 ms 5,376 KB
testcase_36	AC	2 ms 5,376 KB
testcase_37	AC	2 ms 5,376 KB
testcase_38	AC	1 ms 5,376 KB
testcase_39	AC	2 ms 5,376 KB
testcase_40	AC	2 ms 5,376 KB
testcase_41	AC	2 ms 5,376 KB
testcase_42	AC	1 ms 5,376 KB
testcase_43	AC	2 ms 5,376 KB
testcase_44	AC	1 ms 5,376 KB
testcase_45	AC	2 ms 5,376 KB
testcase_46	AC	2 ms 5,376 KB
testcase_47	AC	1 ms 5,376 KB
testcase_48	AC	2 ms 5,376 KB
testcase_49	AC	1 ms 5,376 KB
testcase_50	AC	1 ms 5,376 KB
testcase_51	AC	1 ms 5,376 KB
testcase_52	AC	2 ms 5,376 KB
testcase_53	AC	1 ms 5,376 KB
testcase_54	AC	2 ms 5,376 KB
testcase_55	AC	1 ms 5,376 KB
testcase_56	AC	1 ms 5,376 KB
testcase_57	AC	1 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include "bits/stdc++.h"
using namespace std;
#ifdef _DEBUG
#include "dump.hpp"
#else
#define dump(...)
#endif

#define int long long
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rrep(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define all(c) begin(c),end(c)
const int INF = sizeof(int) == sizeof(long long) ? 0x3f3f3f3f3f3f3f3fLL : 0x3f3f3f3f;
const int MOD = (int)(1e9) + 7;
template<class T> bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; }
template<class T> bool chmin(T &a, const T &b) { if (b < a) { a = b; return true; } return false; }

#define double long double
const double EPS = 1e-8;

// 二次方程式を解の公式を用いて解く
// 計算量: O(sqrt()の計算量)
// 重解は1つにせず返す 1つにしたいなら変えること
// Verified:
//   https://www.urionlinejudge.com.br/judge/en/runs/code/13486899
//   http://codeforces.com/contest/233/submission/52143945
vector<double> quadraticEquation(double a, double b, double c) {
	// 1次方程式
	if (abs(a) < EPS)
		return abs(b) < EPS ? vector<double>() : vector<double>{ -c / b };
	// 判別式
	double D = b * b - 4 * a*c;
	// 解無し
	if (D < -EPS)
		return vector<double>();
	// 重解
	if (D < EPS)
		return vector<double>{-b / (2 * a), -b / (2 * a)};
	double d = sqrt(D);
	double x1 = (-b + d) / (2 * a);
	double x2 = (-b - d) / (2 * a);
	if (x1 > x2)swap(x1, x2);
	return vector<double>{x1, x2};
}

// 二分法
// 方程式の求根アルゴリズム
// f(l) < 0 かつ 0 < f(r) の場合に限る
double bisectionMethod(double l, double r, function<double(double)> f) {
	double yl = f(l), yr = f(r);
	if (yl > yr)swap(l, r);
	assert((yl - EPS) * (yr + EPS) <= 0);
	for (int i = 0; i < 200; i++) {
		double m = (l + r) / 2;
		if (f(m) > 0)
			r = m;
		else
			l = m;
	}
	return l;
}

// 三次方程式
// 二分法でまず1つの解pを求める
// a(x-p)(x^2+sx+t) と式変形して二次方程式で残りの解を求める
vector<double> cubicEquation(double a, double b, double c, double d) {
	if (abs(a) < EPS)
		return quadraticEquation(b, c, d);
	double p = bisectionMethod(-INF, INF, [&](double x) {return a * x*x*x + b * x*x + c * x + d; });
	double t = -d / a / p, s = (c / a - t) / (-p);
	auto ret = quadraticEquation(1, s, t);
	ret.push_back(p);
	sort(ret.begin(), ret.end());
	return ret;
}

signed main() {
	cin.tie(0);
	ios::sync_with_stdio(false);
	int a, b, c; cin >> a >> b >> c;
	auto w = cubicEquation(1, a, b, c);
	sort(all(w));
	vector<int> v;
	rep(i, -1, 2) {
		rep(j, 0, 3) {
			int x = w[j] + i;
			if (x*x*x + a * x*x + b * x + c == 0)
				v.push_back(x);
		}
	}
	sort(all(v));
	v.erase(unique(v.begin(), v.end()), v.end());
	cout << v[0]; rep(i, 1, v.size()) { cout << " " << v[i]; } cout << endl;
	return 0;
}

yukicoder

結果

テストケース

ソースコード