最小的L2损失

小张用某种仪器获得了一组观测值A，A长度为n，A中每一个数都是0-255之间的整数。但是观测值A存在一定的误差，现在小张想求得最可能的真实值B，使得：

• B中的每一个数都是0-255的整数且与A长度相同
• 0< = B(i+1) -B(i) <= 3( 0 <= i < n-1)
• A和B之间的L2损失最小（A(i)和B(i)差的平方的和最小）
  要求返回最小的L2损失。

输入第一行是数据长度n(0<n<1000)
第二行是n个0-255之间的整数
返回一个整数表示L2损失

例子：

输入：
5
1 2 3 6 10

返回
1

c++实现

#include<vector>
#include<stdlib.h>
#include<stdio.h>
#include<iostream>
#include <math.h>

using namespace std;

int minLoss(int n, vector<int> A)
{
    int sumLoss = 0;
    vector<int> B(n, 0);//初始化B
    B[0] = A[0];
    for (int i = 1; i < A.size(); i++)
    {
        if (A[i] - B[i - 1] > 3)
        {
            B[i] = B[i-1] + 3;
            sumLoss += pow(A[i] - B[i],2);
        }
        else if (A[i] - B[i - 1] < 0)
        {
            B[i] = B[i - 1];
            sumLoss += pow(A[i] - B[i], 2);
        }
        else
        {
            B[i] = A[i];
        }
    }
    return sumLoss;
}

int main()
{
    int n;
    cin >> n;//第一个输入是数据长度n

    vector<int> A;
    for (int i = 0; i < n; i++)
    {
        int temp;
        scanf_s("%d", &temp);//依次输入观测值A
        A.push_back(temp);
    }

    int res = minLoss(n, A);
    printf("min L2 loss is : %d", res);

    system("pause");
    return 0;
}

示例1
运行结果 示例2
运行结果 示例3 运行结果