wikioi1180 模拟

最新推荐文章于 2021-02-16 10:37:28 发布

原创最新推荐文章于 2021-02-16 10:37:28 发布 · 547 阅读

0 ·

CC 4.0 BY-SA版权

codevs(wikioi) 同时被 2 个专栏收录

22 篇文章

订阅专栏

模拟

14 篇文章

订阅专栏

本文分享了一段使用C++解决特定问题的程序代码。该程序通过预处理指令、命名空间引入等方式，实现了一个能够计算特定条件下组合数量的算法。通过对输入数据的巧妙处理和中间结果的有效累计，有效地解决了问题。

#include<cstdio>
#include<cstring>
#include <algorithm>
#include <iostream>
#define SUM(x) (x+n)
using namespace std;
int n,b,k,num[50000],sum[50000],lft[100000],rht[100000],ans;
int main()
{
while(~scanf("%d%d",&n,&b))
{
ans=0;
memset(num,0,sizeof(num));
memset(sum,0,sizeof(sum));
memset(lft,0,sizeof(lft));
memset(rht,0,sizeof(rht));
for(int i=1;i<=n;i++)
{
scanf("%d",&num[i]);
if(num[i]==b)
{
k=i;
num[i]=0;
}
else num[i]=num[i]>b?1:-1;
}
lft[SUM(0)]=rht[SUM(0)]=1;
for(int i=k-1;i>=1;i--)
{
sum[i]=sum[i+1]+num[i];
lft[SUM(sum[i])]++;
}
for(int i=k+1;i<=n;i++)
{
sum[i]=sum[i-1]+num[i];
rht[SUM(sum[i])]++;
}
for(int i=1-n;i<=n-1;i++)
ans+=lft[SUM(i)]*rht[(n<<1)-SUM(i)];
printf("%d\n",ans);
}

return 0;
}