TOJ 3896 Above the Median -- 分治树状数组

最新推荐文章于 2021-01-12 17:03:50 发布

randygx

最新推荐文章于 2021-01-12 17:03:50 发布

阅读量634

点赞数

分类专栏： ACM 文章标签：分治树状数组

本文链接：https://blog.csdn.net/randygx/article/details/26008855

版权

ACM 专栏收录该内容

56 篇文章 0 订阅

订阅专栏

题目链接：http://acm.tzc.edu.cn/acmhome/problemdetail.do?&method=showdetail&id=3896

题目大意：给定一个只有1和-1的数列，求连续非负子序列的个数。（当然原题不是这么直白的，要适当转化一下）

分析：很困。。。脑袋不清醒。。。完全忘了树状数组（这个写起来很简单）。于是纠结之下用分治做的，时间复杂度和树状数组是一样的。

假设现在求数列a[s..t]中的非负子序列个数，记为f(s, t)。令mid = (s + t) / 2，f(s, t) = f(s, mid) + f(mid + 1, t) + 包含a[mid]和a[mid+1]的子序列个数。为了计算第三部分，先统计a[mid+1..t]的前缀和出现的次数，用t[i]表示和不超过i的总数，再计算a[s..mid]的后缀和，若某个和为sum，那么有(t - mid - t[sum - 1])个满足条件的子序列。分治时间复杂度可以表示为：

f(n) = 2f(n / 2) + O(n)

也就是f(n) = nlogn，和树状数组一样。绕了好大一圈啊。。。。。

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
         #include 
         
           #include 
          
            #include 
           
             #include 
            
              #include 
             
               #include 
              
                #include 
               
                 #include 
                
                  #include 
                 
                   #include 
                  
                    #include 
                   
                     #define mp make_pair #define X first #define Y second #define MEMSET(a, b) memset(a, b, sizeof(a)) using namespace std; typedef unsigned int ui; typedef long long ll; typedef unsigned long long ull; typedef pair 
                    
                      pii; typedef vector 
                     
                       vi; typedef vi::iterator vi_it; typedef map 
                      
                        mii; typedef priority_queue 
                       
                         pqi; typedef priority_queue 
                        
                          , greater 
                         
                           > rpqi; typedef priority_queue 
                          
                            pqp; typedef priority_queue 
                           
                             , greater 
                            
                              > rpqp; const int MAX_N = 100000 + 2; const int HALF = 50000 + 1; int a[MAX_N]; int sum[MAX_N]; int t[MAX_N]; ll cnt(int lft, int rht) { if (lft == rht) { return a[lft] == 1 ? 1 : 0; } int mid = (lft + rht) >> 1; ll ret = cnt(lft, mid) + cnt(mid + 1, rht); int i, j; for (i = HALF - mid + lft - 2; i <= HALF + rht - mid; ++i) { t[i] = 0; } sum[mid] = 0; for (i = mid + 1; i <= rht; ++i) { sum[i] = sum[i - 1] + a[i]; ++t[sum[i] + HALF]; } for (i = HALF - rht + mid + 1; i <= HALF + rht - mid; ++i) { t[i] += t[i - 1]; } assert(t[HALF + rht - mid] == rht - mid); sum[mid + 1] = 0; for (i = mid; i >= lft; --i) { sum[i] = sum[i + 1] + a[i]; ret += rht - mid - t[HALF - 1 - sum[i]]; } return ret; } int main(int argc, char *argv[]) { // freopen("D:\\in.txt", "r", stdin); int n, x, i; cin >> n >> x; for (i = 1; i <= n; ++i) { scanf("%d", a + i); a[i] = a[i] >= x ? 1 : -1; } printf("%I64d\n", cnt(1, n)); return 0; }

randygx

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
TOJ 3896 Above the Median -- 分治树状数组

题目链接：http://acm.tzc.edu.cn/acmhome/problemdetail.do?&method=showdetail&id=3896题目大意：给出一w
复制链接

扫一扫