Luogu P4889 kls与flag 【思维/排序】By cellur925

题目传送门

这题真的一点也不难qwq。只要想出来就没有什么代码难度的qwq。

每个竹竿只可能向左倒或向右倒，把这两种情况都存在数组中，将数组排序，就可以知道最后落在同一位置的有多少竹竿。就可以知道落在这个位置的竹竿对数，注意，比如有k个竹竿落在同一位置，那么是有k+k-1+k-2+...+1对的。

开longlong，没了。

Code

#include<cstdio>
#include<algorithm>

using namespace std;
typedef long long ll;

int n,m,tot;
ll tmp,ans;
int seq[400090];

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        int x=0;
        scanf("%d",&x);
        seq[++tot]=i-x;
        seq[++tot]=i+x;
    }
    sort(seq+1,seq+1+tot);
    for(int i=1;i<=tot;i++)
    {
        if(seq[i]!=seq[i-1]) tmp=0;
        else tmp++,ans+=tmp;
    }
    printf("%lld",ans);
    return 0;
}