前缀和的妙用

最新推荐文章于 2024-06-22 22:10:32 发布

多行不译必自闭

最新推荐文章于 2024-06-22 22:10:32 发布

阅读量180

点赞数

分类专栏：数据结构文章标签：前缀和

本文链接：https://blog.csdn.net/dajiangyou123456/article/details/97644062

版权

数据结构专栏收录该内容

15 篇文章 0 订阅

订阅专栏

2019暑期牛客多校第4场-K.number
code

#include<iostream>
#include<string.h>
using namespace std;

const int MAXN=1e5+10;
char s[MAXN];
long long cot[4];
int sum[MAXN];
int main()
{
	memset(cot,0,sizeof(cot));
	scanf("%s",s+1);
	int len=strlen(s+1);
	sum[0]=0;
	long long ans=0;
	for(int i=1;i<=len;i++)
	{
		if(s[i]=='0') ans++; 
		sum[i]=(s[i]-'0'+sum[i-1])%3;
		if(s[i]=='0'&&s[i-1]=='0')
		{
			ans+=cot[sum[i]];
		}
		cot[sum[i-1]]++;
		
	}

	
	cout<<ans<<endl;
	
}

牛客小白月赛16-D.小阳买水果
思路
核心思想是前面一个前缀和加上一个大于0的数得到的后面的前缀和大于该前缀和，为保证跨度最大，所以对于遍历的每个前缀和从最右边找大于他的前缀和。找到即返回其下标L，L-i为该前缀和后面能合法买到水果的最大区间，挨个遍历，在取一下max,买水果的最大区间自然就出来了
code

#include<iostream>
#include<cmath>
#include<string.h>
using namespace std;
const int MAXN=2*1e6+10;
long long sum[MAXN];
long long tr[4*MAXN];
void build(int L,int R,int id)
{
    if(L==R)
    {
        tr[id]=sum[L];
        return;
    }
    int m=(L+R)>>1;
     build(L,m,id<<1);
     build(m+1,R,(id<<1)+1);
    tr[id]=max(tr[id<<1],tr[(id<<1)+1]);
    return;
}
int query(int L,int R,int id,int zh )
{
    if(L==R)
    {      
         return L;
    }
    int m=(L+R)>>1;
    if(zh<tr[(id<<1)+1]) return query(m+1,R,(id<<1)+1,zh);
    if(zh<tr[id<<1]) return query(L,m,id<<1,zh);
    return -1;
}
int main()
{
    int n,a;
    cin>>n;
//    memset(sum,0,sizeof(sum));
    for(int i=1;i<=n;i++)
    {
        cin>>a;
        sum[i]=sum[i-1]+a;
    }
    build(1,n,1);
    int ans=0;
    if(sum[1]>0) ans=1;
    for(int i=0;i<n;i++)
    {
        int ww=query(1,n,1,sum[i]);
        if(ww==-1) continue;
        ans=max(ans,ww-i);
    }
    cout<<ans<<endl;
}