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<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define ll long longusing 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 (st<a[i][0])        {            while (a[i][ed]-a[i][st]<=k && ed<=a[i][0]) ed++;            st++;            sum+=ed-st;        }        ans+=sum;    }    cout<<ans<<endl;    return 0;}