1. 程式人生 > >STL Map kls與flag

STL Map kls與flag

題意:地上有一排 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;
}