Задание 27. Информатика. ЕГЭ. Поляков-2675

Имеется набор данных, состоящий из положительных целых чисел. Необходимо определить количество пар элементов \((a_i, \, a_j)\) этого набора, в которых \(1 \leqslant i+5 \leqslant j \leqslant N\) и сумма элементов кратна \(14\).

Входные данные. Даны два входных файла (файл \(A\) и файл \(B\)), каждый из которых содержит в первой строке количество чисел \(N\) \((1 \leqslant N \leqslant 100000)\). Каждая из следующих \(N\) строк содержит одно натуральное число, не превышающее \(10~000\).

Пример входного файла:

\(8\)
\(7\)
\(5\)
\(6\)
\(12\)
\(24\)
\(7\)
\(9\)
\(12\)

Для указанных входных данных количество подходящих пар должно быть равно \(2\). В приведённом наборе имеются две пары \((7, \, 7)\) и \((5, \, 9)\), сумма элементов которых кратна \(14\) и индексы в последовательности отличаются не менее, чем на \(5\).

В ответе укажите два числа: сначала количество подходящих пар для файла \(А\), затем для файла \(B\).

#include<iostream>
#include<fstream>
#include<string>
#include<vector>

using namespace std;

int32_t main() {
    string base = "";
    string fnames[] = {"27-A.txt", "27-B.txt"};

    for(string fn: fnames) {
        ifstream in(base + fn);
        int N, p = 0;
        in >> N;

        vector<int> v(N), r(14, 0);
        for (int i = 0; i < N; i++)
            in >> v[i];

        for (int i = 0; i < N - 5; i++) {
            int t = v[i + 5];
            r[v[i] % 14]++;
            p += r[t % 14 ? 14 - t % 14 : 0];
        }

        cout << p << '\n';
    }
}