algolab/src/deck_of_cards.cpp at main · dsparber/algolab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#include <iostream>
#include <vector>

using namespace std;

void solve() {
  int n; cin >> n;
  int k; cin >> k;
  int vi;
  vector<int> v;
  for (int i = 0; i < n; ++i) {
    cin >> vi;
    v.push_back(vi);
  }
  int i = 0, j = 0;
  int bestI = 0, bestJ = 0;
  int best = 1073741834;
  int currentSum = 0;
  int value;
  while (j < n) {

    currentSum += v[j];
    value = abs(k - currentSum);

    while (i <= j) {
      if (abs(k - (currentSum - v[i])) < value) {
        currentSum -= v[i];
        ++i;
        value = abs(k - currentSum);
      }
      else {
        break;
      }
    }

    if (value < best) {
      best = value;
      bestI = i;
      bestJ = j;
    }
    ++j;
  }
  cout << bestI << " " << bestJ << endl;
}

int main() {
  int t; cin >> t;
  for (int i = 0; i < t; ++i) {
    solve();
  }
}