[LGOJ] 题解 P4889 【kls与flag】

洛谷月赛加赛的题

点击跳题

点击跳洛谷blog

$\text{点个赞,谢谢}$

思路大概如下(使用STL的map)

• 定义函数$f(x)$,代表$1+2+...+x-1$的和

---

1. 输入竹竿高度$k$ ,$map[k]++$,所以$map[k]$表示k处有几根竹竿顶重合
2. 遍历map,对于$ans=ans+f(i\text{的值})$

---

对于$f(x)$,不难理解,一个点有x个竿重合,两两配对,显然是$(1+2+...+x-1)$次

map用来统计哪些点被几根竹竿覆盖,(如数据较小可以用数组)

---

话不多说，上代码!

#include<bits/stdc++.h>
using namespace std;
long long n,m; //注意开long long,否则30分
map<long long,long long> mmp; //定义map
long long val(long long x) //f(x)
{
	if(x<2) return 0; //竹竿不能盖在一起
	return x*(x-1)/2; //等差数列公式
}
int main()
{
	cin>>n>>m; //输入
	for(long long i=1;i<=n;i++)
	{
		long long k;
		scanf("%lld",&k);
		mmp[k+i]++;	//向右倒竹竿顶落在的点
		mmp[i-k]++;	//向左倒竹竿顶落在的点
	}
	long long ans=0; //计数器
    
    //遍历map
	for(map<long long,long long>::iterator  i=mmp.begin();i!=mmp.end();i++) 
	{
    	//i->second表示该点有（i->second）个竿重合
	    ans+=val(i->second);//统计
	}
	cout<<ans;//输出
	return 0;
}