“蔚来杯“2022牛客暑期多校赛第二场 J-Link with Arithmetic Progression

最新推荐文章于 2024-10-09 22:28:58 发布

问不问童子

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量229

点赞数

分类专栏：问不问题解之旅文章标签：算法

本文链接：https://blog.csdn.net/weixin_69232161/article/details/126023358

版权

等差数列线性回归最小误差数学建模 C++实现

关键词由CSDN通过智能技术生成

问不问题解之旅专栏收录该内容

7 篇文章 0 订阅

订阅专栏

题目大意：构造一个等差数列a'，使得 $\sum$ $(ai-a'i)^{2}$ 最小

由高中数学知识可知该题可转化成求 $(i,ai)$ 各点的线性回归方程，以求出原式的最小值

线性回归方程求法

指路↓↓↓↓↓↓↓↓↓↓

https://blog.csdn.net/chehec2010/article/details/90045835https://blog.csdn.net/chehec2010/article/details/90045835公式：

$\hat{y}=bx+a$

$b=\frac{n\sum_{i=1}^{n}x_{i}y_{i}-(\sum_{i=1}^{n}x_{i})(\sum_{i=1}^{n}y_{i})}{n\sum_{i=1}^{n}x_{i}^{2}-(\sum_{i=1}^{n}x_{i})}$

$a=\bar{y}-b\bar{x}$

#include "stdio.h"
#include "math.h"
#include "stdlib.h"
#include "string.h"
#include "iostream"
#include "algorithm"

using namespace std;

const int N=100005;
long long a[N];

namespace GTI
{
    char gc(void)
       {
        const int S = 1 << 16;
        static char buf[S], *s = buf, *t = buf;
        if (s == t) t = buf + fread(s = buf, 1, S, stdin);
        if (s == t) return EOF;
        return *s++;
    }
    long long gti(void)
	{
        long long a = 0, b = 1, c = gc();
        for (; !isdigit(c); c = gc()) b ^= (c == '-');
        for (; isdigit(c); c = gc()) a = a * 10 + c - '0';
        return b ? a : -a;
    }
}
using GTI::gti;

int main()
{
	long long T;
	T=gti();
	while(T--)
	{
		long long n;
		n=gti();
		for(long long i=1;i<=n;i++)
			a[i]=gti();
		double x,y=0;
		x=(1+n)*n*1.0/2/n;
		for(long long i=1;i<=n;i++)
			y+=a[i];
		y/=n;
		double b1=0,b2=0;
		for(long long i=1;i<=n;i++)
			b1+=(i-x)*(a[i]-y),b2+=(i-x)*(i-x);
		b1/=b2;
		double aa=y-b1*x,ans=0;
		for(long long i=1;i<=n;i++)
			ans+=(a[i]-b1*i-aa)*(a[i]-b1*i-aa);
		printf("%.15lf\n",ans);
	}
	return 0;
}

读入使用了题目给的快读，因为精度问题所有int都开了long long