zoj 3963 heap partion

Posted Omz

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了zoj 3963 heap partion相关的知识,希望对你有一定的参考价值。

https://vjudge.net/problem/ZOJ-3963

题意:

给出一个数列,可以用这个数列构造一种二叉树,这个二叉树满足数的下标 i <= j,并且 si <= sj,si是sj的父亲,问给出的数列可以构造多少棵这样的二叉树。

思路:

这题赛上没有写出来,看了题解之后给补的。

首先,通过这题学到了,memset初始化数组有时是会造成超时的。set的upper_bound(x)这个函数,它返回set中大于x的第一个元素的位置(注意是大于,不是大于等于)。

于是,这题就是贪心加set。贪心指的是对于当前的输入的x,在前面找到小于等于它的数,如果说没有找到,那么就把这个数插入,作为一棵新的树的根。如果说找到了,那么就把这个点插入这棵树,并且把这个数可插入的数量减1,当数量为0的时候就不能再插入了。具体看看注释。

代码:

  1 #include <stdio.h>
  2 #include <string.h>
  3 #include <vector>
  4 #include <set>
  5 #include <algorithm>
  6 using namespace std;
  7 
  8 int num,node[100005],cnt[100005];//node保存节点x的编号,cnt数组表示每个节点能插入的剩下的位置
  9 
 10 vector<int> ans[100005];//ans保存答案
 11 set<int> s;
 12 
 13 void solve(int x,int pos)
 14 {
 15     set<int>::iterator it;
 16 
 17     it = s.upper_bound(x);
 18 
 19     if (it == s.begin())//x是最小的,所以选择插入
 20     {
 21         s.insert(x);
 22 
 23         cnt[x] = 2;//一开始有两个空位
 24 
 25         node[x] = num;
 26 
 27         ans[node[x]].push_back(pos);
 28 
 29         num++;
 30     }
 31     else
 32     {
 33         --it;//因为是大于x的位置,所以要--
 34 
 35         if (x == (*it))//相等,就不用插入,旧的元素
 36         {
 37 
 38             cnt[x]++;//相当于是加了2个位置,然后自己又占了一个
 39 
 40             ans[node[x]].push_back(pos);
 41         }
 42         else
 43         {
 44             node[x] = node[*it];
 45 
 46             cnt[*it]--;
 47 
 48             if (cnt[*it] == 0) s.erase(*it);//没有位置了,删除
 49 
 50             cnt[x] = 2;//新的元素
 51 
 52             s.insert(x);//后面插入是防止迭代器改变
 53 
 54             ans[node[x]].push_back(pos);
 55         }
 56     }
 57 }
 58 int main()
 59 {
 60     int t;
 61 
 62     scanf("%d",&t);
 63 
 64     while (t--)
 65     {
 66 
 67         s.clear();
 68 
 69         num = 0;
 70 
 71         int n;
 72 
 73         scanf("%d",&n);
 74 
 75         for (int i = 0;i <= n;i++)
 76         {
 77             node[i] = cnt[i] = 0;
 78             ans[i].clear();//说了n的总和不超过2 * 10 ^ 6,用memeset反而会超时
 79         }
 80 
 81         for (int i = 1;i <= n;i++)
 82         {
 83             int x;
 84 
 85             scanf("%d",&x);
 86 
 87             solve(x,i);
 88         }
 89 
 90         printf("%d\n",num);
 91 
 92         for (int i = 0;i < num;i++)
 93         {
 94             int sz = ans[i].size();
 95 
 96             printf("%d",sz);
 97 
 98             for (int j = 0;j < sz;j++)
 99             {
100                 printf(" %d",ans[i][j]);
101             }
102 
103             printf("\n");
104         }
105     }
106 
107     return 0;
108 }

 

以上是关于zoj 3963 heap partion的主要内容,如果未能解决你的问题,请参考以下文章

ZOJ 3963 Heap Partition(multiset + stl自带二分 + 贪心)题解

bzoj3963--cdq分治

●BZOJ 3963 [WF2011]MachineWorks

bzoj3963[WF2011]MachineWorks cdq分治+斜率优化dp

BZOJ3963[WF2011]MachineWorks cdq分治+斜率优化

Partioner的理解