E. Special Elements(前缀和)

题目链接
题意:
给你一个数组,如果其中一个数是连续的两个或更多的数的和,那么我们称这个数为特殊数,求数组中有多少个特殊数。
思路:
利用前缀和思想,再遍历做差,用一个flag数组标记做差的值。
代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e6+5;
const int inf=0x3f3f3f3f;
signed main()
{
    IOS;
    int t;
    cin>>t;
    while(t--)
    {
        int n,arr[10005],brr[10005],mp[8005],mx=0,sum=0,num=0;
        memset(mp,0,sizeof mp);
        cin>>n;
        for(int i=0;i<n;i++)
        {
            cin>>arr[i];
            sum+=arr[i];
            brr[i]=sum;
            mx=max(arr[i],mx);
        }
        for(int i=0;i<n;i++)
        {
            if(i>0&&brr[i]<=mx)
            {
                mp[brr[i]]=1;
            }
            for(int j=i+2;j<n;j++)
            {
                if(brr[j]-brr[i]<=mx)
                {
                    mp[brr[j]-brr[i]]=1;
                }
                else
                {
                    break;
                }
            }
        }
        for(int i=0;i<n;i++)
        {
            if(mp[arr[i]]==1)
            {
                num++;
            }
        }
        cout<<num<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/ACkingdom/article/details/106306194
今日推荐