STL Map kls与flag

题意：地上有一排$n$根竹竿，竹竿的间距均为一个单位长度，高度在$1\sim m$之间。每根竹竿可以往左倒或者往右倒。如果两根竹竿在选择方向放倒之后，它们的顶端可以重合，那么称它们是优秀的。求有多少对竹竿是优秀的。$n<=2e5，m<=1e9$

做法比较简单，$sum[i]$表示所有竹竿向左或向右倒下后竹竿头落在$i$这个位置的数量，对于点$i$，有$sum[i]$个竹竿头落在这里，即有$C_n^2$优秀的竹竿。可是$m$的大小为$1e9$，我们开不下这么大的数组怎么办？这里就使用了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;
}