版权声明:未经本蒟蒻同意,请勿转载本蒻博客 https://blog.csdn.net/wddwjlss/article/details/83146048
题意:地上有一排 根竹竿,竹竿的间距均为一个单位长度,高度在 之间。每根竹竿可以往左倒或者往右倒。如果两根竹竿在选择方向放倒之后,它们的顶端可以重合,那么称它们是优秀的。求有多少对竹竿是优秀的。
做法比较简单, 表示所有竹竿向左或向右倒下后竹竿头落在 这个位置的数量,对于点 ,有 个竹竿头落在这里,即有 优秀的竹竿。可是 的大小为 ,我们开不下这么大的数组怎么办?这里就使用了STL的Map,Map可以看做一个下表可以为各种东西(long long,string等)的数组,最后遍历Map求解即可,遍历的方法类似Set。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
#define ll long long
using namespace std;
ll a[200200],ans;
map<ll,int> sum;
int n,m;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
sum[i-a[i]]++;
sum[i+a[i]]++;
}
for(map<ll,int>::iterator it=sum.begin();it!=sum.end();it++)
{
int cnt=it->second;
ans+=1ll*cnt*(cnt-1)/2;
}
printf("%lld",ans);
return 0;
}