CSES - Datatähti 2024 alku - Results
Submission details
Task:Säähavainnot
Sender:stpn129
Submission time:2023-11-12 14:53:40 +0200
Language:C++ (C++11)
Status:READY
Result:53
Feedback
groupverdictscore
#1ACCEPTED52.75
Test results
testverdicttimescore
#1ACCEPTED0.02 s6.75details
#2ACCEPTED0.02 s7details
#3ACCEPTED0.02 s6.75details
#4ACCEPTED0.02 s6.38details
#5ACCEPTED0.02 s6.75details
#6ACCEPTED0.02 s6.25details
#7ACCEPTED0.02 s6.38details
#8ACCEPTED0.02 s6.5details

Code

#include <bits/stdc++.h>
 
using namespace std;
 
void init_code() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
#ifndef ONLINE_JUDGE
  freopen("input.txt", "r", stdin);
  freopen("output.txt", "w", stdout);
#endif
}
 
pair<double, double> best_fitting_line(vector<pair<double, double>>& v) {
    int n = v.size();
    
    double sum_x = 0, sum_x2 = 0, sum_y = 0, sum_xy = 0, a, b;
    for (int i = 0; i < n; ++i) {
        sum_x = sum_x + v[i].first;
        sum_x2 = sum_x2 + v[i].first * v[i].first;
        sum_y = sum_y + v[i].second;
        sum_xy = sum_xy + v[i].first * v[i].second;
    }
 
    a = (n * sum_xy - sum_x * sum_y) / (n * sum_x2 -sum_x * sum_x);
    b = (sum_y - a * sum_x) / n;
 
    return {a, b};
}
 
void solve() {
    int n = 24;
    vector<double> a(n);
    vector<pair<double, double>> b;
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        if (i > 20) {
            b.emplace_back(i, a[i]);
        }
    }
 
 
    mt19937 mersenne(static_cast<unsigned int>(time(0)));
    uniform_real_distribution<> rng(-0.3, 0.4);
 
    pair<double, double> predict = best_fitting_line(b);
 
    for (int i = 0; i < 12; ++i) { 
        double res = (predict.first * (n + i) + predict.second + rng(mersenne)); 
        res += 0.9845 * a[n - 1 - i];
        res /= 2;
        if (i > 4) {
            cout << '?' << ' ';     
        } else {
            cout << res  << ' ';
        }
    }
}
 
signed main() {
  init_code();
  int t = 1;
  cin >> t;
  while (t--) {
    solve();
  }
  return 0;
}

Test details

Test 1

Verdict: ACCEPTED

input
1000
-0.4 -0.1 -0.2 -0.3 -0.4 -0.5 ...

correct output
0.4 0.4 0.5 0.8 0.9 1.1 1.3 1....

user output
0.179949 0.435507 0.214017 0.0...
Truncated

Test 2

Verdict: ACCEPTED

input
1000
2.9 2.9 2.9 2.1 2.6 2 2 2.2 2....

correct output
2.3 1.6 1.5 1.1 1 0.7 0.6 0.8 ...

user output
2.56135 2.74346 2.59619 2.5445...
Truncated

Test 3

Verdict: ACCEPTED

input
1000
6.6 6 6.4 6 4.6 4.6 4.2 4.3 4....

correct output
10 10.9 10.3 10.1 9.1 7.3 5.7 ...

user output
10.4032 10.5181 10.4512 10.233...
Truncated

Test 4

Verdict: ACCEPTED

input
1000
19.4 20.2 19.1 18.9 18.3 17.3 ...

correct output
18 18.2 17 17.5 17.2 16.2 12 8...

user output
17.2149 17.6689 17.252 16.9308...
Truncated

Test 5

Verdict: ACCEPTED

input
1000
-5.7 -5.8 -5.8 -5.9 -7.1 -6.9 ...

correct output
-4.2 -4.1 -4 -3.8 -3.5 -3.2 -3...

user output
-4.45029 -4.24318 -4.41467 -4....
Truncated

Test 6

Verdict: ACCEPTED

input
1000
14.8 14.8 15.4 12.9 11.8 9.7 9...

correct output
11.8 11 11.6 10.8 10.4 10.4 10...

user output
13.1172 13.9212 13.1652 13.735...
Truncated

Test 7

Verdict: ACCEPTED

input
1000
0.7 1 2 1.4 0.6 -0.4 -0.9 -0.7...

correct output
-1.3 -0.5 -0.6 -1 -3.2 -7.2 -6...

user output
-1.77122 -1.56411 -1.7356 -1.8...
Truncated

Test 8

Verdict: ACCEPTED

input
1000
15.1 15.3 14.9 14.4 14.4 13.7 ...

correct output
15.6 15.9 16 15.2 14.6 14.4 13...

user output
14.8819 14.8194 14.9199 13.983...
Truncated