函数

题目

题目描述
给出 $n$ 个二次函数 $f_i(x)=a_i{x}^2+b_{i}x+c_i(i\in [1,n])$ 。求一组正整数 $x_i(i\in [1,n])$ ，满足 ${\sum }_{i=1}^n{x}_i=m$ 且 ${\sum }_{i=1}^n{f}_i(x_i)$ 最小。

输出 ${\sum }_{i=1}^n{f}_i(x_i)$ 即可。

输入格式
第一行是两个正整数 $n,m$ ；接下来 $n$ 行，第 $i+1$ 行是三个整数 $a_i,b_i,c_i$（含义见题面）。

输出格式
共一行，一个整数，表示 ${\sum }_{i=1}^n{f}_i(x_i)$ 。

数据范围与约定

数据点编号	$m$
$1,2$	$\le 10$
$3,4$	$\le 100$
$5,6$	$\le 500$
$7,8,9,10$	$\le 5000$
$\forall x\in [11,20]$	$\le 10^5$

对于所有数据满足：

• $n\le m,a_i\in [1,10^3]$
• $b_i,c_i\in [-10^3,10^3]$

思路

暴力。考虑每次将某个 $x$ 增大 $1$ 。直接用堆即可。时间复杂度 $\mathcal{O}(m\log n)$ 。

---

$2019/12/1\mathtt{u}\mathtt{p}\mathtt{d}\mathtt{a}\mathtt{t}\mathtt{e}$

原来我根本没有意识到正确性是由 $a>0$ 保证的……

一个 $a>0$ 的二次函数，开口向上，令 $g(x)=f(x+1)-f(x)$ ，一定有 $g(x)$ 单增。从导数的角度很好说明。

所以，如果你不将 $g(x)$ 最小的 $x_i$ 增加，其余的 $g(x)$ 一定恒大于该 $g(x)$，一定不优。

代码

#include <cstdio>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
inline int readint(){
	int a = 0, f = 1; char c = getchar();
	for(; c<'0' or c>'9'; c=getchar())
		if(c == '-') f = -1;
	for(; '0'<=c and c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(long long x){
	if(x < 0) putchar('-'), x = -x;
	if(x > 9) writeint(x/10);
	putchar(x%10+'0');
}

const int MaxN = 100000;
int n, m, a[MaxN], b[MaxN], c[MaxN];
inline long long function(int i,long long x){
	return a[i]*x*x+b[i]*x+c[i];
}
inline long long addv(int i,long long x){
	return function(i,x+1)-function(i,x);
}
struct Status{
	int id, x;
	Status(int I,int X):id(I),x(X){ }
	bool operator<(const Status &that)const{
		return addv(id,x) > addv(that.id,that.x);
	}
};
priority_queue<Status> pq;

int main(){
	n = readint(), m = readint();
	for(int i=0; i<n; ++i){
		a[i] = readint();
		b[i] = readint();
		c[i] = readint();
	}
	long long ans = 0;
	for(int i=0; i<n; ++i){
		ans += function(i,1);
		pq.push(Status(i,1));
	}
	for(int i=n; i<m; ++i){
		auto p = pq.top(); pq.pop();
		ans += addv(p.id,p.x ++);
		pq.push(p);
	}
	writeint(ans), putchar('\n');
	return 0;
}

---

延伸题目

题目描述
$n$个二次函数 $f_i=a_i{x}^2+b_{i}x+c_i(i\in [1,n])$，求一组 实数 $x_i(i\in [1,n])$，满足 ${\sum }_{i=1}^n{x}_i=m$ 且 ${\sum }_{i=1}^n{f}_i(x_i)$ 最小。

输出 ${\sum }_{i=1}^n{f}_i(x_i)$ 即可。

输入格式
第一行，一个正整数 $n$ 和一个 非负实数 $m$；接下来 $n$ 行，第 $i+1$ 行是三个 实数 $a_i,b_i,c_i$（含义见题面）。

输出格式
共一行，一个 实数 ，表示 ${\sum }_{i=1}^n{f}_i(x_i)$ 。保留两位小数。

数据范围与约定
对于所有数据都有：

• $n,m\le 10^5,a_i\in (0,10^3]$
• $b_i,c_i\in [-10^3,10^3]$

思路

推广到实数域之后，使用堆来存储的方法就不适用了。

现在该怎么办呢？——虽然存不下，我们还是应该考虑一下原来的方法。

考虑每次将一个 $x$ 变大 $\Delta x(\Delta x\to 0)$ 。变化量，不就是 $f^{\prime }(x)\Delta x$ 吗？而且，因为 ${\lim }_{\Delta x\to 0}{f}^{\prime }(x-\Delta x)=f^{\prime }(x)$ ，所以无论增大、减小，变化量都是 $f^{\prime }(x)\Delta x$ 。

所以，最优情况下，$f^{\prime }(x)$ 应该全部相等。否则，将 $f^{\prime }(x)$ 较小的 $x$ 增加 $\Delta x$ 、较大的减少 $\Delta x$ ，就会得到更优的解。

解方程 $f_i^{\prime }(x_i)=f_j^{\prime }(x_j)$ ，即 $2a_i{x}_i+b_i=2a_j{x}_j+b_j$，得到

$$x_i=\frac{2a_j{x}_j+b_j-b_i}{2a_i}$$

将每一个 $x_i(i\ne 1)$ 与 $x_1$ 建立关系（即，用带 $x_1$ 的式子表示 $x_i$ ），就可以通过 ${\sum }_{i=1}^n{x}_i=m$ 解出 $x_1$ 。然后一一还原即可！

---

$2019/12/1\mathtt{u}\mathtt{p}\mathtt{d}\mathtt{a}\mathtt{t}\mathtt{e}$

会不会无解呢？数据保证了 $a>0$，所以 $x_i=\frac{a_1}{a_i}{x}_1+\frac{b_1-b_i}{2a_i}$ 中的系数$\frac{a_1}{a_i}$为正，所以总和也为正，故一定有解。而且 $a>0$ 也可以保证答案不会变成 $-\infty$.

---

时间复杂度 $\mathcal{O}(n)$ 。竟然比整数域的情况还要快

代码

然而并没有代码，因为根本没有这道题，这是我自己口胡的。所以题解也是我自己 $yy$ 的。有没有错误呢？我确实不知道。请大佬指正！