Codeforces Round #640 (Div. 4) E. Special Elements

题目链接

思路：要求我们找连续区间和能否组成一个数组元素，因为数据范围比较小，我们用前缀和去算区间和，然后map保存每个数组元素出现次数，暴力寻找每种区间和的值，判断他是否在数组中出现过。

#include 
#include 
#include 
#include 
#include

//#include
using namespace std;
typedef long long ll;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
const int mod=1e4+7;
const int N=1e5+10;
const int inf=0x7f7f7f7f;


ll gcd(ll a,ll b)
{
    return b==0?a:gcd(b,a%b);
}

ll lcm(ll a,ll b)
{
    return a*(b/gcd(a,b));
}

template <class T>
void read(T &x)
{
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
        if(c == '-')
            op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
        x = x * 10 + c - '0';
    if(op)
        x = -x;
}
template <class T>
void write(T x)
{
    if(x < 0)
        x = -x, putchar('-');
    if(x >= 10)
        write(x / 10);
    putchar('0' + x % 10);
}
int a[8006],mp[8006],sum[8006];


int main()
{
    SIS;
    int t;
    cin>>t;
    while(t--)
    {
        memset(mp,0,sizeof mp);
        int n;
        cin>>n;
        for(int i=1;i<=n;i++)
        {
            cin>>a[i];
            mp[a[i]]++;
            sum[i]=sum[i-1]+a[i];
        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                int tm=sum[j]-sum[i-1];
                if(tm<=n&&mp[tm])
                {
                    ans+=mp[tm];
                    mp[tm]=0;
                }
            }
        }
        cout<<ans<<endl;


    }




    return 0;
}