洛谷 P2309 loidc,卖卖萌

P2309 loidc,卖卖萌

题目背景

Loidc萌萌哒。

他最近一直在靠卖萌追求他的真爱——vivym，经过几轮攻势后vivym酱眼看就要被他所攻略。擅长数据结构的vivym决定利用强大的数据结构之力帮助自己逃脱。

题目描述

Vivym给了loidc一串阿拉伯数字。他将一个数串的每个数字之和定义为该数串的数串和，vivym要求loidc求出数字中有多少个子串的数串和为正数。他想了半天总是解决不好，于是他便将这一棘手的问题交给了编程能力极强的你，他希望你能在1s内解决。不然vivym就要闪现过墙幽灵疾步跑掉了

输入输出格式

输入格式：

第一行一个数n，表示数串的长度。

第二行一共n个数，它就是那串阿拉伯数字

输出格式：

一个数，表示铭文中有多少个子串的数串和为正数。

输入输出样例

输入样例#1： 复制

3
8 -9 2

输出样例#1： 复制

3

说明

30% n<=1000。

100% n<=100000。

对于题目读入的其他变量均介于-100~100。

思路：前置和+逆序对。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n;
long long ans;
long long num[],sum[],tmp[];
void up(int l,int mid,int r){
    int i=l,j=mid+,tot=;
    while(i<=mid&&j<=r){
        if(num[i]>num[j])    ans+=mid-i+;
        tmp[++tot]=num[i]<=num[j]?num[i++]:num[j++];
    }
    while(i<=mid)    tmp[++tot]=num[i++];
    while(j<=r)    tmp[++tot]=num[j++];
    for(int k=;k<=tot;k++)
        num[l++]=tmp[k];
}
void work(int l,int r){
    if(l<r){
        int mid=(l+r)/;
        work(l,mid);
        work(mid+,r);
        up(l,mid,r);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=;i<=n;i++){
        scanf("%lld",&num[i]);
        sum[i]=sum[i-]+num[i];
    }
    for(int i=;i<=n;i++){
        num[i]=sum[n-i+];
        if(num[i]>)    ans++;
    }
    work(,n);
    cout<<ans;
}