좋은 친구

시간 제한1초메모리 제한128 MB

요약
등수 순서로 주어진 학생 이름들에서 등수 차이가 K 이하이고 이름 길이가 같은 학생 쌍의 개수를 구합니다.
난이도

보통10점 중 4점

유형
슬라이딩 윈도우, 배열, 구현
정답자
아직 제출이 없습니다

문제

N명의 학생 이름이 성적순으로 주어진다. 두 학생의 등수 차이가 K 이하이고, 두 학생 이름의 길이가 같으면 두 학생은 좋은 친구이다.

좋은 친구인 학생 쌍의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 주어진다. (3 <= N <= 300,000, 1 <= K <= N)

다음 N개 줄에는 학생의 이름이 성적순으로 하나씩 주어진다. 이름은 알파벳 대문자로만 이루어져 있으며, 길이는 2 이상 20 이하이다.

출력

첫째 줄에 좋은 친구인 학생 쌍의 수를 출력한다.

예제2

  1. 예제 1

    입력
    4 2
    IVA
    IVO
    ANA
    TOM
    
    예상 출력
    5
    
  2. 예제 2

    입력
    6 3
    CYNTHIA
    LLOYD
    STEVIE
    KEVIN
    MALCOLM
    DABNEY
    
    예상 출력
    2