Säähavainnot

#include <iostream>
#include <vector>
#include <iomanip>

// Simple linear regression model
class LinearRegression {
public:
    double slope;
    double intercept;

    LinearRegression() : slope(0.0), intercept(0.0) {}

    // Fit the model using provided data
    void fit(const std::vector<double>& x, const std::vector<double>& y) {
        double sum_x = 0.0;
        double sum_y = 0.0;
        double sum_x2 = 0.0;
        double sum_xy = 0.0;
        int n = x.size();

        for (int i = 0; i < n; ++i) {
            sum_x += x[i];
            sum_y += y[i];
            sum_x2 += x[i] * x[i];
            sum_xy += x[i] * y[i];
        }

        slope = (n * sum_xy - sum_x * sum_y) / (n * sum_x2 - sum_x * sum_x);
        intercept = (sum_y - slope * sum_x) / n;
    }

    // Predict the value for a given x
    double predict(double x) const {
        return slope * x + intercept;
    }
};

int main() {
    int n;
    std::cin >> n;

    for (int day = 0; day < n; ++day) {
        std::vector<double> pastTemperatures(24);
        for (int i = 0; i < 24; ++i) {
            std::cin >> pastTemperatures[i];
        }

        LinearRegression model;
        std::vector<double> x, y;

        // Prepare the data for regression
        for (int i = 0; i < 12; ++i) {
            x.push_back(i);
            y.push_back(pastTemperatures[i]);
        }

        // Fit the model
        model.fit(x, y);

        // Predict the temperatures for the next 12 hours
        for (int i = 12; i < 24; ++i) {
            double prediction = model.predict(i);
            if (i == 23) {
                std::cout << std::fixed << std::setprecision(1) << prediction << "\n";
            } else {
                std::cout << std::fixed << std::setprecision(1) << prediction << " ";
            }
        }
    }

    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED