题意 : 构造尽可能少的一种结构,父节点的值小于等于子节点,而且子节点在序列中出现在父节点后面。而且每个父节点至多有两个子节点。
分析 : 贪心即可,首先,我们先对这个序列排个序,得知每个数的rank 然后对原序列从左到右遍历,维护一个set,存的是这个数的rank, 对于每个数,如果set空,那么直接insert,如果不空,用lower_bound 找出这个数之前的最大的小于等于他的数,如果这个数不存在,也是直接insert ,存在,用一个并查集连边,因为每个节点最多连两条边,那么我们用一个cnt数组记录每个点连了几个子节点,当大于等于2的时候直接从set中删掉。最后答案,个数就是并查集跟自己连边的个数。最后输出即可。
赛后听说被卡时间了,,大概运气比较好,,没有体会到,,回来仔细看了下,题目说n的总和不会大于2e6,那么用memset会超时,直接for1~n清空能过。
以下是代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<cstring>
#include<string>
#include<vector>
#include<iomanip>
//#include<unordered_set>
//#include<unordered_map>
#include<cmath>
#include<list>
#include<bitset>
using namespace std;
#define _____ ios::sync_with_stdio(false); cin.tie(0);
#define ull unsigned long long
#define ll long long
#define lson l,mid,id<<1
#define rson mid+1,r,id<<1|1
typedef pair<int, int>pii;
typedef pair<ll, ll>pll;
typedef pair<ll, int>pli;
typedef pair<double, double>pdd;
const double eps = 1e-6;
const int MAXM = 5005;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int INF = 0x3f3f3f3f;
const double FINF = 1e18;
const ll MOD = 1000000007;
const double PI = acos(-1);
struct lx {
int id, v;
}a[100005], b[100005];
int cmp(lx a, lx b)
{
if (a.v == b.v)return a.id < b.id;
else return a.v < b.v;
}
int pre[100005];
int rk[100005];// rk 记录第 i 大的数 原id 是多少
int rk2[100005];// rk2 记录 原序列 下标为 id 的 排序后是第几位
int cnt[100005];
vector<int>ans[100005];
vector<int>v;
set<int>st;
int find(int x) { return pre[x] == x ? x : pre[x] = find(pre[x]); }
int main()
{
int T, n;
scanf("%d", &T);
while (T--)
{
st.clear();
scanf("%d", &n);
for (int i = 1; i <= n; ++i)scanf("%d", &a[i].v), a[i].id = i, b[i].id = a[i].id, b[i].v = a[i].v;
sort(b + 1, b + 1 + n, cmp);
for (int i = 1; i <= n; ++i)rk[i] = b[i].id, rk2[b[i].id] = i, pre[i] = i, cnt[i] = 0;
for (int i = 1; i <= n; ++i)
{
if (st.size() == 0)
{
st.insert(rk2[i]);
continue;
}
auto pos = st.lower_bound(rk2[i]);
if (pos == st.begin() && *st.begin() > rk2[i])
{
st.insert(rk2[i]);
continue;
}
else
{
int tmp = *(--pos);
pre[i] = rk[tmp];
cnt[tmp]++;
if (cnt[tmp] >= 2)st.erase(tmp);
st.insert(rk2[i]);
}
}
for (int i = 1; i <= n; ++i)ans[i].clear();
v.clear();
for (int i = 1; i <= n; ++i)
{
if (pre[i] == i)v.push_back(i);
ans[find(i)].push_back(i);
}
printf("%d\n", v.size());
for (int i = 0; i < v.size(); ++i)
{
sort(ans[v[i]].begin(), ans[v[i]].end());
printf("%d ", ans[v[i]].size());
for (int j = 0; j < ans[v[i]].size(); ++j)
{
if (j == ans[v[i]].size() - 1)printf("%d\n", ans[v[i]][j]);
else printf("%d ", ans[v[i]][j]);
}
}
}
}