COCI '12 Contest 3 #3 Malcolm

View as PDF

Submit solution


Points: 7
Time limit: 1.0s
Memory limit: 32M

Problem types

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.

Input Specification

The first line of input contains two positive integers, N (3 \le N \le 300\,000) and K (1 \le K \le 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.

Output Specification

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

Comments

There are no comments at the moment.