====== Codeforces Round #699 (Div. 2) ====== [[https://codeforces.com/contest/1481|比赛链接]] ===== E. Sorting Books ===== ==== 题意 ==== 给定一个长度为 $n$ 的序列 $\{a\}$,每次操作可以任取一个位置并将该位置的数移到序列末尾。 问最少需要多少次操作才能使序列中所有大小相同的数相邻。 ==== 题解 ==== 显然只需要在序列中选定一些位置,然后按特定顺序将他们移除就可以完成任务。定义 $\text{dp}(i)$ 表示不选择序列 $[i,n]$ 中的位置个数的最大值。 维护每个值 $v$ 的最靠左出现的位置 $l_v$,最靠右出现的位置 $r_v$,动态维护 $[i,n]$ 中出现的次数 $c_v$。 如果保留位置 $i$ 。当 $i=l_{a_i}$ 时,需要移除 $[i,r_{a_i}]$ 中所有不等于 $a_i$ 的位置,即 $\text{dp}(i)\gets c_{a_i}+\text{dp}(r_{a_i}+1)$。 当 $i\neq l_{a_i}$ 时,若不保留 $l_{a_i}$ 的位置,则 $l_{a_i}$ 的位置的数会被加到位置 $n$ 后面。 为了使得 $l_{a_i}$ 的位置的数与 $i$ 位置的数所在段相邻,需要移除 $[i,n]$ 中所有不等于 $a_i$ 的位置,于是 $\text{dp}(i)\gets c_{a_i}$。 若保留 $l_{a_i}$ 的位置,则该情况会在计算 $\text{dp}(l_{a_i})$ 时考虑,此时不考虑。 如果不保留位置 $i$,则 $\text{dp}(i)\gets \text{dp}(i+1)$。最终答案为 $n-\text{dp}(1)$。时间复杂度 $O(n)$。 const int MAXN=5e5+5; int a[MAXN],dp[MAXN],l[MAXN],r[MAXN],c[MAXN]; int main() { int n=read_int(); _rep(i,1,n)a[i]=read_int(); for(int i=n;i;i--){ l[a[i]]=i; if(!r[a[i]])r[a[i]]=i; } for(int i=n;i;i--){ c[a[i]]++; dp[i]=dp[i+1]; if(i==l[a[i]])dp[i]=max(dp[i],dp[r[a[i]]+1]+c[a[i]]); else dp[i]=max(dp[i],c[a[i]]); } enter(n-dp[1]); return 0; } ===== F. AB Tree ===== ==== 题意 ==== 给定一棵以 $1$ 为根节点的有根树以及 $x$ 个字符 $a$ 和 $n-x$ 个字符 $b$。要求用所给字符对树上每个结点进行标注。 给定每个节点代表一个字符串,该字符串由根节点到该节点路径上的所有字符依次拼接而成。 要求最小化所有节点代表的字符串中的互异的字符串个数,同时输出任意一种方案。 ==== 题解 ==== 设树的深度为 $d$,根节点深度为 $0$。于是答案一定不小于 $d+1$,因为一定有长度为 $1\sim d+1$ 的字符串。 先考虑如何判定答案是否为 $d+1$ 以及如何构造方案。为了满足条件,必须满足每个长度的字符串只有一种,于是同一深度的节点字符一定相同。 不妨设深度 $i$ 共有 $c_i$ 个节点,于是只需要从 $c_0,c_1\cdots c_d$ 中选出若干个数构成 $x$ 即可。 这是一个背包 $\text{dp}$。考虑优化方案,首先由于 $\sum_{i=0}^d c_i=n$,所以 $c_i$ 最多只有 $O(\sqrt n)$ 个互异的值。 考虑将所有值相同的 $c_i$ 放在一起考虑,本题转化为多重背包问题,且物品权值只有 $O(\sqrt n)$ 种。 设 $\text{dp}(i,j)$ 表示只考虑前 $i$ 种权值物品是否存在权值和为 $j$ 的方案。 设当前物品权值为 $\text{val}$,数量为 $\text{cnt}$。于是对 $\text{dp}(i,j)$,只需要找到 $0\le k\le \text{cnt}$ 使得 $\text{dp}(i-1,j-k*\text{val})=1$ 即可。 不妨记 $k_j$ 表示满足上述条件的最小的 $k$,且如果不存在 $k$ 满足条件则 $k_j=-1$。 于是有若 $\text{dp}(i-1,j)=1$ 则 $k_j=0$,否则 $k_j=k_{j-\text{val}}+1$,注意 $k_{j-\text{val}}=-1$ 和 $k_{j-\text{val}}+1\gt \text{cnt}$ 的情况。 于是可以 $O(n)$ 维护每种权值,$O(n\sqrt n)$ 判定答案是否为 $d+1$。 如果答案等于 $d+1$,则逆序暴力并输出方案,时间复杂度 $O(n)$。 ps. 据说二进制优化 + $\text{bitset}$ 可以做到 $O\left(\frac {n\sqrt n}w\right)$ 否则,答案一定为 $d+2$,下面给出构造。 首先假设当前层还剩下 $x$ 个字符 $a$ 和 $y$ 个字符 $b$,当前节点数为 $c_i$。 若 $c_i$ 满足 $c_i\le \max(x,y)$,则直接将该层节点用一种字符覆盖即可。 否则,考虑当前层的非叶子节点数。剩下的节点数总和为 $x+y$,而非叶子结点数总和一定不大于叶子结点数总和。 于是当前层的非叶子节点数 $\le$ 叶子结点数总和 $\le \frac {x+y}2\le \max(x,y)$。 不妨设 $\max(x,y)=x$,于是用全部 $a$ 覆盖当前层所有非叶子节点与部分叶子节点,其他剩余节点全部用 $b$ 覆盖,于是总答案仅 $+1$。 总时间复杂度 $O(n\sqrt n)$。 const int MAXN=1e5+5,MAXM=500; struct Edge{ int to,next; }edge[MAXN]; int head[MAXN],edge_cnt; void AddEdge(int u,int v){ edge[++edge_cnt]=Edge{v,head[u]}; head[u]=edge_cnt; } int sz[MAXN],last[MAXN],maxd; vector node[MAXN],c[MAXN],cv; bool dp[MAXM][MAXN]; char ans[MAXN]; void dfs(int u,int d){ node[d].push_back(u); sz[u]=1,maxd=max(maxd,d); for(int i=head[u];i;i=edge[i].next){ int v=edge[i].to; dfs(v,d+1); sz[u]+=sz[v]; } } bool cmp(int u,int v){ return sz[u]>sz[v]; } int main() { int n=read_int(),x=read_int(); _rep(i,2,n) AddEdge(read_int(),i); dfs(1,0); _for(i,0,n) c[node[i].size()].push_back(i); _rep(i,1,n){ if(c[i].size()>0) cv.push_back(i); } dp[0][0]=true; _rep(i,1,cv.size()){ int val=cv[i-1],cnt=c[val].size(); mem(last,-1); _for(j,0,val){ if(dp[i-1][j]) dp[i][j]=true,last[j]=0; else dp[i][j]=false,last[j]=-1; } _rep(j,val,n){ if(dp[i-1][j]) dp[i][j]=true,last[j]=0; else if(last[j-val]!=-1&&last[j-val]+1<=cnt) dp[i][j]=true,last[j]=last[j-val]+1; else dp[i][j]=false,last[j]=-1; } } if(dp[cv.size()][x]){ enter(maxd+1); _rep(i,1,n)ans[i]='b'; for(int i=cv.size(),j=x;i;i--){ int val=cv[i-1],pos=0; while(!dp[i-1][j]){ j-=val; int d=c[val][pos++]; _for(k,0,val) ans[node[d][k]]='a'; } } puts(ans+1); } else{ enter(maxd+2); pair p1('a',x),p2('b',n-x); _for(i,0,n){ if(p1.second