STL Map kls與flag
阿新 • • 發佈:2018-11-08
題意:地上有一排 根竹竿,竹竿的間距均為一個單位長度,高度在 之間。每根竹竿可以往左倒或者往右倒。如果兩根竹竿在選擇方向放倒之後,它們的頂端可以重合,那麼稱它們是優秀的。求有多少對竹竿是優秀的。
做法比較簡單, 表示所有竹竿向左或向右倒下後竹竿頭落在 這個位置的數量,對於點 ,有 個竹竿頭落在這裡,即有 優秀的竹竿。可是 的大小為 ,我們開不下這麼大的陣列怎麼辦?這裡就使用了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;
}