STL Map kls与flag

版权声明:未经本蒟蒻同意,请勿转载本蒻博客 https://blog.csdn.net/wddwjlss/article/details/83146048

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

做法比较简单, s u m [ i ] sum[i] 表示所有竹竿向左或向右倒下后竹竿头落在 i i 这个位置的数量,对于点 i i ,有 s u m [ i ] sum[i] 个竹竿头落在这里,即有 C n 2 C^2_n 优秀的竹竿。可是 m m 的大小为 1 e 9 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;
}

猜你喜欢

转载自blog.csdn.net/wddwjlss/article/details/83146048