kls与flag

题目描述:

有n个OI选手，每个人插了一个flag。有一天因为某种原因，flag都被触发了，所以地上有一排n根竹竿，竹竿的间距均为一个单位长度，高度在1~m之间。

kls看到这些竹竿，感觉不好看，于是准备把它们全部放倒。

在这之前，kls想到了一个数学问题。每根竹竿可以往左倒或者往右倒。如果两根竹竿在选择方向放倒之后，它们的顶端可以重合，那么称它们是优秀的。现在kls想要知道，有多少对竹竿是优秀的。

输入格式:

第一行两个数n,m，表示竹竿的个数和最大高度。

第二行n个正整数，表示每根竹竿的高度。

输出格式

输出一行，只有一个数，表示有多少对竹竿是优秀的。

样例:

样例输入 #1

5 5

2 3 3 3 2

样例输出 #1

3

提示

样例解释:

- 1号和2号向左倒可以顶端重合
- 4号和5号向右倒可以顶端重合
- 1号向右，5号向左可以顶端重合

数据范围:

对于30%的数据，满足n≤2000，m≤5000；

对于60%的数据，满足n≤200000，m≤500000；

对于100%的数据，满足n≤200000,m10^9。

---

分析：

这题，其实比较简单，第一想法是枚举，看每个点的落点能不能重合，简单是简单，但你先看看时间同意不，其实就是TLE。

那该咋优化？？  其实将每个flag的落点统计进一个数组，为了省时间加个sort排序，然后进行判断前后是否一致，是的话答案增加，但注意不要一次就加1，为啥呢，这就像组合问题。

比如有3个flag都有一个共同的落点，分别命名为n1,n2,n3。

n1和n2一组，n1和n3一组，n2和n3一组，一共三组，如果就前后判断就会少，那么我们可以发现1+2+...+n的规律，但这么做又复杂了，那我们定义个l，将n的个数加进答案里，当然是前后相同的情况下，否则清空。

注意：这题数据大，变量定义成long long,输入和输出分别用scanf("%lld"),printf("%lld"),%lld是long long 用的。

代码不就来了。

---

代码：

#include <bits/stdc++.h>
using namespace std;
long long a[200002],b[400004];
int main()
{
    long long n,m,ans=0,len=0;
	scanf("%lld%lld",&n,&m);//输入
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		len+=2;//因为可以往左或右倒
		b[len]=i+a[i],b[len-1]=i-a[i];//记录倒的点
	}
	sort(b+1,b+1+len);//排序别忘
	int k=0;
	for(int i=1;i<=2*n;i++)//一共有2*n个点
	{
		if(b[i]!=b[i-1]) k=0;
		else             k++,ans+=k;
	}
	printf("%lld",ans);//输出
    return 0;
}

别抄，我相信你会。 

---

 看完了，你学会了吗？

验证方法：洛谷P4889 kls与flag直接AC

有问题可以在评论区讨论