zoj 3963 heap partion

  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 }