Malcolm
Ограничения: время – 1s/2s, память – 64MiB Ввод: input.txt или стандартный ввод Вывод: output.txt или стандартный вывод
Послать решение Blockly Посылки Темы Где Обсудить (0)
Since teacher Herkabe has started ranking his N students, the number of friendships in his class has
sharply fallen. The students near the bottom of the rankings list have become jealous of the top
students, while the top students started looking down on their less successful colleagues.
According to Malcolm's observations, the following rule holds: two students are friends if their ranks
are close enough, more precisely, if they differ by at most `K`. For example, if `K\ =\ 1`, then only
neighbouring students on the rankings list are friends. Furthermore, two students are good friends if
they are friends and their names have the same length.
Write a program to calculate the number of pairs of good friends in this gifted class.
The first line of input contains two positive integers, `N` (`3\ ≤\ N\ ≤\ 300\ 000`) and `K` (`1\ ≤\ \ K\ ≤\ N`), from
the problem statement.
Each of the following `N` lines contains a single student's name. The names are given in the order they
appear on the rankings list. They consist of between 2 and 20 (inclusive) uppercase English letters.
The first and only line of output must contain the required number of pairs.
Sample Input #1
4 2
IVA
IVO
ANA
TOM
Sample Input #2
6 3
CYNTHIA
LLOYD
STEVIE
KEVIN
MALCOLM
DABNEY
Source: COCI 2012/2013, contest #3