2016暑期集训14A找朋友

问题 A: 找朋友

时间限制: 1 Sec 内存限制: 128 MB

题目描述
随着N名新生开始入学，会产生很多对好朋友。今年新生成为好朋友的方式有些特别。他们会看中成绩排名和彼此之间的名字。

给定所有学生的姓名和成绩排名，和一个整数K。两人能成为朋友，首先两人的成绩排名差距至多为K。例如K=1，那么只有成绩相邻的才能成为朋友。其次，名字长度相同的才能成为朋友。

输入
第一行两个正整数，N(3<=N<=300000)和K(1<=K<=N)，分别表示人数和朋友间排名差距。

接下来N行，每行表示新生姓名，给出的顺序就是按照成绩排名。每个人的姓名长度属于范围[2,20]。

输出
输出新生中可以产生的朋友对数。

样例输入
4 2
IVA
IVO
ANA
TOM
样例输出
5
提示
【数据规模】

对于20%的数据，N ≤ 20

对于40%的数据，N ≤ 30000

对于这道题首先想到的自然是暴力N*K
但显然时间不允许，所以我们必须用更快的方法。
观察发现,交朋友要名字一样长,而且数据不大
所以分2-20组,每组作一个类似于队列的线扫,统计数量即可.

另外,考试中只得了80分,原因是我写了st < ed,而数据中有运行过程中st=ed的情况,请lwq大神帮助debug,改为st < a[i,0]就对了

#include
#include
#include
#include
#define ll long long
using namespace std;
int n,k;
char s[25];
int a[25][300005];
int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        int len=strlen(s+1);
        a[len][++a[len][0]]=i;
    }
    ll ans=0;
    for (int i=2;i<=20;i++)
    {
        int st=1,ed=2;
        ll sum=0;
        while (st0])
        {

            while (a[i][ed]-a[i][st]<=k && ed<=a[i][0]) ed++;
            st++;
            sum+=ed-st;
        }
        ans+=sum;
    }
    cout<return 0;
}