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

题目传送门

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

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

开longlong,没了。

Code

 1 #include<cstdio>
 2 #include<algorithm>
 3 
 4 using namespace std;
 5 typedef long long ll;
 6 
 7 int n,m,tot;
 8 ll tmp,ans;
 9 int seq[400090];
10 
11 int main()
12 {
13     scanf("%d%d",&n,&m);
14     for(int i=1;i<=n;i++)
15     {
16         int x=0;
17         scanf("%d",&x);
18         seq[++tot]=i-x;
19         seq[++tot]=i+x;
20     }
21     sort(seq+1,seq+1+tot);
22     for(int i=1;i<=tot;i++)
23     {
24         if(seq[i]!=seq[i-1]) tmp=0;
25         else tmp++,ans+=tmp;
26     }
27     printf("%lld",ans);
28     return 0;
29 }
View Code

 

posted @ 2018-09-24 09:24  cellur925&Chemist  阅读(163)  评论(0编辑  收藏  举报