题目描述
一共有 n 个小朋友排成一排,身高分别是 a1,a2,⋯,an。每一个小朋友都想找到与自己距离不少于 k 且身高相同的小朋友玩。请你找到所有希望在一起玩的小朋友。
距离的定义是,两个小朋友所站在的位置的数值之差。
输入格式
第一行两个整数 n,k。
第二行 n 个整数 a1,a2,⋯,an。
输出格式
输出若干行,每行两个整数 i,j,表示 i 号小朋友和 j 号小朋友希望在一起玩。你需要注意,总是保证 i<j,并且 i 小的排前面,i 相同的就 j 小的排前面。
最后一行一个整数,表示希望在一起玩的小朋友的数量。
样例
样例输入
6 3
1 3 3 2 1 3
样例输出
1 5
2 6
3 6
3
提示
对于100%的数据,保证0<=k<=n<=2000。
思路
这道题数据并不是很大,直接模拟也不会超时,所以我们直接按照题意模拟即可。
大致的概念就是:
首先输入数组,接着按照要求进行一个一个的找朋友,一边找一边输出,并且累加,最后再将累加的结果输出就可以了
注意:我觉得一边找一边输出更好一点,一个是省时间,不用存起来再输出,另一个则是可以防止将顺序搞乱。
代码
#include<iostream>
using namespace std;
int n,k,ans;
int a[3000];
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
{
int x=i+k;
if(x<=n)
{
for(int j=x;j<=n;j++)
{
if(a[j]==a[i])
{
cout<<i<<" "<<j<<endl;
ans++;
}
}
}
}
cout<<ans;
return 0;
}
后记
点个赞吧!