【模板】三分 (P3382)

【模板】三分

题目描述

如题，给出一个 $N$ 次函数，保证在范围 $[l,r]$ 内存在一点 $x$，使得 $[l,x]$ 上单调增，$[x,r]$ 上单调减。试求出 $x$ 的值。

输入格式

第一行一次包含一个正整数 $N$ 和两个实数 $l,r$，含义如题目描述所示。

第二行包含 $N+1$ 个实数，从高到低依次表示该 $N$ 次函数各项的系数。

输出格式

输出为一行，包含一个实数，即为 $x$ 的值。若你的答案与标准答案的相对或绝对误差不超过 $10^{-5}$ 则算正确。

样例 #1

样例输入 #1

3 -0.9981 0.5
1 -3 -3 1

样例输出 #1

-0.41421

提示

对于 $100\%$ 的数据，$6\le N\le 13$，函数系数均在 $[-100,100]$ 内且至多 $15$ 位小数，$|l|,|r|\le 10$ 且至多 $15$ 位小数。$l\le r$。

【样例解释】

如图所示，红色段即为该函数 $f(x)=x^3-3x^2-3x+1$ 在区间 $[-0.9981,0.5]$ 上的图像。

当 $x=-0.41421$ 时图像位于最高点，故此时函数在 $[l,x]$ 上单调增，$[x,r]$ 上单调减，故 $x=-0.41421$，输出 $-0.41421$。

三分法的思路

取区间内两个三等分点 midL （表示左三等分点）、 midR （表示右三等分点）。
对于上图，当 midL 所对应的函数值 大于 midR 所对应的函数值时，只有两种情况：一种是极值点在 midL 和 midR 的中间，另一种是极值点在 midL 和 midR 的右边。可见无论哪种情况， midL 都在极值点左边。故当 midL 所对应的函数值大于 midR 所对应的函数值时，让子区间的左端点取 midL ，从而缩小区间。
当 midL 所对应的函数值 小于 midR 所对应的函数值时，与上面类似，只有两种情况：一种是极值点在 midL 和 midR 的中间，另一种是极值点在 midL 和 midR 的左边。可见无论哪种情况， midR 都在极值点右边。故当 midL 所对应的函数值小于 midR 所对应的函数值时，让子区间的右端点取 midR ，从而缩小区间。

代码

#include <iostream>
using std::cin, std::cout, std::ios_base;
const double eps = 1e-6; // 精度为 10 的 -6 次方
int n; // 函数的最高次项的次数
double a[15]; // 函数由高到低次项的系数
double f_x(double x) // 传入参数 x ，返回函数值
{
	double res = 0.0; // 用于保存结果
	for (int i = 0; i <= n; i++) // 函数由 n + 1 项组成
		res = res * x + a[i]; // 可以换成 res *= x, res += a[i];
	return res; // 返回结果
}
int main()
{
	double left, right; // 用于保存区间的左、右端点值
	cin >> n >> left >> right;
	for (int i = 0; i <= n; i++) // 函数由 n + 1 项组成
		cin >> a[i];
	double trisection, midL, midR; // 分别为区间长度的三等分、区间的左三等分点、区间的右三等分点
	while (right - left > eps) // 直到区间长度小于精度才退出循环
	{
		trisection = (right - left) / 3.0;
		midL = left + trisection, midR = right - trisection;
		if (f_x(midR) > f_x(midL)) // 如果右三等分点的函数值大于左三等分点的函数值
			left = midL; // 则让子区间的左端点取左三等分点
		else
			right = midR; // 否则让子区间的右端点取右三等分点
	}
	// 以下两行代码用于控制输出精度为小数点后 5 位
	cout.precision(5);
	cout.setf(ios_base::fixed, ios_base::floatfield);
	cout << left; // 最终区间的左端点就是答案
	return 0;
}