print1936. Malcolm

printMalcolm

Ограничения: время – 1s/2s, память – 64MiB Ввод: input.txt или стандартный ввод Вывод: output.txt или стандартный вывод copy
Послать решение 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 Output #1

5

Sample Input #2

6 3
CYNTHIA
LLOYD
STEVIE
KEVIN
MALCOLM
DABNEY

Sample Output #2

2
Source: COCI 2012/2013, contest #3
loading