( ext{Strange Queries})
解法
首先有这样的转移:
其中 (0/1) 表示 (n) 否/是与 (n-1) 建立关系。但这题需要我们进行区间合并,这个 (mathtt{dp}) 提醒我们从两个方向来思考:维护 (mathtt{dp}) 转移方式(矩阵);改写一个可以区间合并的 (mathtt{dp})。
这里只讲区间合并的 (mathtt{dp})。类似地,我们在区间 ([l,r]) 上维护四个变量:(l,r) 两个端点都建立关系的最小值;(l,r) 两个端点中,(l) 或 (r) 可以不 建立关系的最小值;(l,r) 均 可以不 建立关系的最小值。这四个变量是一层一层想到的。另外,需要注意 "可以不" 并不是 "不",这导致转移时可以不考虑某些情况,因为已经被包含在限制较小的变量中了。
具体就用一棵平衡树维护,还需要维护区间 (max,min) 来表示 (lvert a_n-a_{n-1} vert)。最开始建树的时候要先排序。
代码
#include <cstdio>
#define print(x,y) write(x),putchar(y)
template <class T>
inline T read(const T sample) {
T x=0; char s; bool f=0;
while((s=getchar())>'9' or s<'0')
f|=(s=='-');
while(s>='0' and s<='9')
x=(x<<1)+(x<<3)+(s^48),
s=getchar();
return f?-x:x;
}
template <class T>
inline void write(const T x) {
if(x<0) {
putchar('-');
write(-x);
return;
}
if(x>9) write(x/10);
putchar(x%10^48);
}
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int n,q,idx,rt;
struct node {
int mn,mx,siz,ls,rs,key,val;
ll ans,bl,br,ba;
void Put() {
printf("VALUE %d %d %d
",mn,mx,val);
printf("%lld %lld %lld %lld
",ba,bl,br,ans);
}
} t[maxn];
int NewNode(int x) {
t[++idx].siz=1;
t[idx].ls=t[idx].rs=0;
t[idx].key=rand();
t[idx].siz=1;
t[idx].val=t[idx].mn=t[idx].mx=x;
t[idx].ba=t[idx].bl=t[idx].br=0;
t[idx].ans=1e15;
return idx;
}
void Update(const node &a,const node &b,node &o) {
node t=o; int inc=b.mn-a.mx;
t.ba=min(a.bl+b.br,a.ba+b.ba+inc);
t.bl=min(a.bl+b.ans,a.ba+b.bl+inc);
t.br=min(a.ans+b.br,a.br+b.ba+inc);
t.ans=min(a.ans+b.ans,a.br+b.bl+inc);
o=t;
}
void pushUp(int o) {
if(!o) return;
int l=t[o].ls,r=t[o].rs;
t[o].siz=t[l].siz+t[r].siz+1;
t[o].mn=t[o].mx=t[o].val;
t[o].ba=t[o].bl=t[o].br=0;
t[o].ans=1e15;
if(l) Update(t[l],t[o],t[o]),t[o].mn=t[l].mn;
if(r) Update(t[o],t[r],t[o]),t[o].mx=t[r].mx;
}
void split(int o,int k,int &x,int &y) {
if(!o) x=y=0;
else {
if(t[o].val<=k)
x=o,
split(t[o].rs,k,t[o].rs,y);
else
y=o,
split(t[o].ls,k,x,t[o].ls);
pushUp(o);
}
}
int merge(int x,int y) {
if(!x or !y) return x|y;
if(t[x].key<t[y].key) {
t[x].rs=merge(t[x].rs,y);
pushUp(x);
return x;
}
else {
t[y].ls=merge(x,t[y].ls);
pushUp(y);
return y;
}
}
int Val[maxn];
int main() {
srand(20210907);
int a,b,c;
for(int T=read(9);T;--T) {
n=read(9),q=read(9);
rt=idx=0;
for(int i=1;i<=n;++i)
Val[i]=read(9);
sort(Val+1,Val+n+1);
for(int i=1;i<=n;++i)
rt=merge(rt,NewNode(Val[i]));
while(q--) {
int x,y;
x=read(9),y=read(9);
if(x==1) {
split(rt,y-1,a,b);
rt=merge(merge(a,NewNode(y)),b);
}
else {
split(rt,y-1,a,b);
split(b,y,b,c);
b=merge(t[b].ls,t[b].rs);
rt=merge(merge(a,b),c);
}
print(t[rt].ans>1e14?0:t[rt].ans,'
');
}
}
return 0;
}
( ext{[TJOI 2013] })拯救小矮人
解法
首先将小矮人按照 (a_i+b_i) 从小到大排序。为什么捏?不妨令 (H-(a_i+b_i)) 是第 (i) 个小矮人的逃生高度,对于相邻小矮人 (i,j),如果我们期望 两个小矮人都逃出去,那么放逃生高度越大的小矮人越优。
但问题是有可能小矮人不逃出去更优,可以令 (dp_i) 为逃出 (i) 个小矮人后可以垒出的最大高度,这是一个 (mathcal O(n^2)) 的 (mathtt{dp})。
更优的做法是反悔贪心。先依次使可以逃出的小矮人逃出,并用优先队列记录它的 (a),到某个无法逃出的小矮人 (i) 时,我们考虑这样操作:找到队列中最大的 (a),如果踢回这个小矮人能使 (i) 逃出,且 (max a>a_i),我们就把这个小矮人 (j) 换成 (i)。
首先 (j) 到 (i) 这段的小矮人肯定是不受影响的,如果 (j) 能够使之间的小矮人逃出就早换了。对于之后的小矮人,垒出的高度变高了,解也就变优了。这是 (mathcal O(nlog n)) 的。
代码
#include <cstdio>
#define print(x,y) write(x),putchar(y)
template <class T>
inline T read(const T sample) {
T x=0; char s; bool f=0;
while((s=getchar())>'9' or s<'0')
f|=(s=='-');
while(s>='0' and s<='9')
x=(x<<1)+(x<<3)+(s^48),
s=getchar();
return f?-x:x;
}
template <class T>
inline void write(const T x) {
if(x<0) {
putchar('-');
write(-x);
return;
}
if(x>9) write(x/10);
putchar(x%10^48);
}
#include <queue>
#include <algorithm>
using namespace std;
const int maxn=2005;
int n;
struct node {
int a,b;
bool operator < (const node &t) const {
return a+b<t.a+t.b;
}
} s[maxn];
priority_queue <int> q;
int main() {
n=read(9);
for(int i=1;i<=n;++i)
s[i]=(node){read(9),read(9)};
sort(s+1,s+n+1);
int h=read(9);
long long sum=0; int ans=0;
for(int i=1;i<=n;++i)
sum+=s[i].a;
for(int i=1;i<=n;++i)
if(sum+s[i].b>=h) {
sum-=s[i].a,++ans;
q.push(s[i].a);
}
else {
if(!q.empty() and sum+q.top()+s[i].b>=h and q.top()>s[i].a) {
sum+=q.top()-s[i].a;
q.pop();
q.push(s[i].a);
}
}
print(ans,'
');
return 0;
}
( ext{[ICPC World Finals 2019] Hobson })的火车
题目描述
给你多棵内向基环树,对于每个点,求出有多少个点走 (le k) 步能走到这个点。
解法
感觉自己的脑子里根本不存在差分这个东西,好几次想用树链剖分硬艹了…
先找出环,( m dfs) 每个环上点的子树,用一个栈动态维护这个节点到根的节点,这样就可以找到走 (k+1) 才能走到的那个祖先。
然后考虑点对环的贡献。考虑破环为链,随便选一个点作为 (rt_0)(链的起点)。如果能贡献整个环,就将 (rt_0) 的差分数组加一,反之计算出它在环上贡献的一段区间 ([l,r]),需要注意有 (l>r) 的情况,这时需要将 (rt_0) 的差分数组加一。
代码
#include <cstdio>
#define print(x,y) write(x),putchar(y)
template <class T>
inline T read(const T sample) {
T x=0; char s; bool f=0;
while((s=getchar())>'9' or s<'0')
f|=(s=='-');
while(s>='0' and s<='9')
x=(x<<1)+(x<<3)+(s^48),
s=getchar();
return f?-x:x;
}
template <class T>
inline void write(const T x) {
if(x<0) {
putchar('-');
write(-x);
return;
}
if(x>9) write(x/10);
putchar(x%10^48);
}
#include <vector>
using namespace std;
const int maxn=5e5+5;
int n,k,f[maxn],ans[maxn];
int vis[maxn],stk[maxn],tp;
int dep[maxn],siz;
vector <int> e[maxn],rt;
void dfs(int u,int root) {
if(!vis[u]) vis[u]=1;
stk[++tp]=u;
if(tp>k+1)
if(stk[tp-k-1]^rt[root])
--ans[stk[tp-k-1]];
for(auto v:e[u]) {
if(vis[v]==2) continue;
dep[v]=dep[u]+1;
dfs(v,root);
if(u^rt[root])
ans[u]+=ans[v];
}
--tp;
if(dep[u]-1<=k) {
if(dep[u]-1+siz>k+1) {
++ans[rt[root]];
--ans[rt[(root+k-dep[u]+2+siz)%siz]];
if((root+k-dep[u]+2+siz)%siz<=root)
++ans[rt[0]];
}
else ++ans[rt[0]];
}
if(rt[root]^u) ++ans[u];
}
void work(int x) {
rt.clear();
while(vis[x]^2) {
if(vis[x])
rt.push_back(x);
++vis[x]; x=f[x];
}
siz=rt.size();
for(int i=0;i<rt.size();++i)
dep[rt[i]]=1,dfs(rt[i],i);
for(int i=1;i<rt.size();++i)
ans[rt[i]]+=ans[rt[i-1]];
}
int main() {
n=read(9),k=read(9);
for(int i=1;i<=n;++i) {
f[i]=read(9);
e[f[i]].push_back(i);
}
for(int i=1;i<=n;++i)
if(!vis[i]) work(i);
for(int i=1;i<=n;++i)
print(ans[i],'
');
return 0;
}
简单题
题目描述
有两个长度为 (K) 的数列 (a,b),满足 (sum a_i=N,sum b_i=M)。
定义权值 (P) 为
你需要求出所有可能数列组合的权值之和。取模 (998244353)。
多组数据,(Tle 100,N,M,Kle 5cdot 10^5)。
解法
定义 (dp_{i,j,k}) 为在第 (i) 位,数列 (a) 数字和为 (j),数列 (b) 数字和为 (k) 的权值之和。那么有:
这是 (mathcal O(n^4k)) 的。具体优化可以枚举 (min{x,y}),后两维的前缀和即可,应该还要再容斥一下,就有 (mathcal O(n^3k)) 了。
我们可以思考权值的组合意义 —— 定义 ({c}) 满足 (forall i,c_ile min{a_i,b_i}),那么对于固定的 ({a},{b}),所有合法的 ({c}) 的个数即为权值!
转化一下,对于每种 ({c}),计算 ({a},{b}) 的种类数。
枚举 (s=sum c_i),为了保证条件成立,可以先令 (c_i=a_i=b_i),再将 (N-s),(M-s) 填入 (K) 个位置(插板法),就有:
注意初始时先将 (1) 填入每一个位置,所以后面两个组合数的上面本来还要加个 (K),就抵消掉了。