首页 > 其他分享 >题解 ABC207F【Tree Patrolling】

题解 ABC207F【Tree Patrolling】

时间:2023-06-14 20:44:20浏览次数:43  
标签:tmp sz 题解 ll Patrolling Tree rep dp lld

挺简单的树上背包,就是有点难写。

设 \({dp}_{u,i,x,y}\) 表示仅考虑 \(u\) 的子树内,有 \(i\) 个节点被控制,\(x\) 为节点 \(u\) 是否有警卫,\(y\) 为节点 \(u\) 是否被控制。(其实所有 \(x=1,y=0\) 的状态都没用,但我懒得管了。)

每个点 \(u\) 的初始值为 \({dp}_{u,0,0,0}={dp}_{u,1,1,1}=1\),其他为 \(0\)。因为初始时这个点的背包大小为 \(1\)(只有自己),分别对应没有警卫和有警卫。

设 \(v\) 为 \(u\) 的儿子,考虑合并 \(v\) 的背包到 \(u\) 上。设 \(u,v\) 的背包大小分别为 \(p,q\)。下面设 \(i\in[0,p],j\in[0,q]\),考虑三种状态分别如何转移:

(一)状态 \(x=0,y=0\)。

此时 \(v\) 上一定没有警卫,不然就会覆盖到 \(u\),因此:

\[{dp}_{u,i+j,0,0}\stackrel{+}{\gets}{dp}_{u,i,0,0}\times({dp}_{v,j,0,0}+{dp}_{v,j,0,1}) \]

(二)状态 \(x=0,y=1\)。

有两种情况:第一种是此前 \(u\) 尚未被控制,直到 \(v\) 有警卫后 \(u\) 才被控制;第二种是此前 \(u\) 已经被控制。因此:

\[{dp}_{u,i+j,0,1}\stackrel{+}{\gets}{dp}_{u,i-1,0,0}\times{dp}_{v,j,1,1}+{dp}_{u,i,0,1}\times({dp}_{v,j,0,0}+{dp}_{v,j,0,1}+{dp}_{v,j,1,1}) \]

注意第一种情况 \(i\) 处要减一,因为 \(u\) 点的贡献需要被算进来。

(三)状态 \(x=1,y=1\)。

这种情况对 \(v\) 没有限制,因此:

\[{dp}_{u,i+j,1,1}\stackrel{+}{\gets}{dp}_{u,i,1,1}\times({dp}_{v,j-1,0,0}+{dp}_{v,j,0,1}+{dp}_{v,j,1,1}) \]

注意这里 \(j\) 处又有一个减一,跟上面同理,\(v\) 点的贡献需要被算进来。

在完成 \(v\) 的背包向 \(u\) 的合并后,我们将 \(p\stackrel{+}{\gets}q\) 即可继续合并下一棵子树。

上面减一处均有数组越界的风险,这里认为所有不合法(负数下标)的状态的值均为 \(0\)。

正确实现的树上背包的复杂度为 \(\mathcal O(n^2)\),原因是两个点只在 LCA 处被合并一次。正确实现树上背包的方法是精细枚举背包大小,也就是上文对 \(p,q\) 的处理。这一点在代码中是通过 \(sz\) 数组实现的。

//By: OIer rui_er
#include <bits/stdc++.h>
#define rep(x,y,z) for(ll x=(y);x<=(z);x++)
#define per(x,y,z) for(ll x=(y);x>=(z);x--)
#define debug(format...) fprintf(stderr, format)
#define fileIO(s) do{freopen(s".in","r",stdin);freopen(s".out","w",stdout);}while(false)
using namespace std;
typedef long long ll;

mt19937 rnd(std::chrono::duration_cast<std::chrono::nanoseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
ll randint(ll L, ll R) {
    uniform_int_distribution<ll> dist(L, R);
    return dist(rnd);
}

template<typename T> void chkmin(T& x, T y) {if(x > y) x = y;}
template<typename T> void chkmax(T& x, T y) {if(x < y) x = y;}

const ll N = 2e3+5, mod = 1e9+7;

ll n, dp[N][N][2][2], tmp[N][2][2], sz[N];
vector<ll> e[N];

void dfs(ll u, ll f) {
    dp[u][0][0][0] = dp[u][1][1][1] = 1;
    sz[u] = 1;
    for(ll v : e[u]) {
        if(v != f) {
            dfs(v, u);
            // debug("%lld -> %lld\n", u, v);
            rep(i, 0, sz[u]+sz[v]) {
                tmp[i][0][0] = 0;
                tmp[i][0][1] = 0;
                tmp[i][1][1] = 0;
            }
            per(i, sz[u], 0) {
                per(j, sz[v], 0) {
                    tmp[i+j][0][0] += dp[u][i][0][0] * (dp[v][j][0][0] + dp[v][j][0][1]);
                    tmp[i+j][0][0] %= mod;
                    if(i >= 1) {
                        tmp[i+j][0][1] += dp[u][i-1][0][0] * dp[v][j][1][1] + dp[u][i][0][1] * (dp[v][j][0][0] + dp[v][j][0][1] + dp[v][j][1][1]);
                        tmp[i+j][0][1] %= mod;
                        tmp[i+j][1][1] += dp[u][i][1][1] * ((j >= 1 ? dp[v][j-1][0][0] : 0) + dp[v][j][0][1] + dp[v][j][1][1]);
                        tmp[i+j][1][1] %= mod;
                    }
                }
            }
            rep(i, 0, sz[u]+sz[v]) {
                dp[u][i][0][0] = tmp[i][0][0];
                dp[u][i][0][1] = tmp[i][0][1];
                dp[u][i][1][1] = tmp[i][1][1];
            }
            sz[u] += sz[v];
            // rep(i, 0, sz[u]) debug("DP[%lld][%lld] = {%lld, %lld, %lld}\n", u, i, dp[u][i][0][0], dp[u][i][0][1], dp[u][i][1][1]);
        }
    }
    // debug("@ %lld\n", u);
    // rep(i, 0, sz[u]) debug("DP[%lld][%lld] = {%lld, %lld, %lld}\n", u, i, dp[u][i][0][0], dp[u][i][0][1], dp[u][i][1][1]);
}

int main() {
    // freopen("debug.log", "w", stderr);
    scanf("%lld", &n);
    rep(i, 1, n-1) {
        ll u, v;
        scanf("%lld%lld", &u, &v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs(1, 0);
    rep(i, 0, n) printf("%lld\n", (dp[1][i][0][0]+dp[1][i][0][1]+dp[1][i][1][1])%mod);
    return 0;
}

标签:tmp,sz,题解,ll,Patrolling,Tree,rep,dp,lld
From: https://www.cnblogs.com/ruierqwq/p/abc207f.html

相关文章

  • jqtreetable jquery-treeview
    jqtreetable[img]http://dl.iteye.com/upload/attachment/466717/80fc34ec-ed82-3c04-b2f4-5de688bbf990.jpg[/img]jquery-treeview[img]http://dl.iteye.com/upload/attachment/466715/3c0521cb-37e0-3fc6-8dfc-f438b48e8569.jpg[/img]......
  • 【题解】[六省联考 2017] 寿司餐厅
    题目描述:Kiana最近喜欢到一家非常美味的寿司餐厅用餐。每天晚上,这家餐厅都会按顺序提供\(n\)种寿司,第\(i\)种寿司有一个代号\(a_i\)和美味度\(d_{i,i}\),不同种类的寿司有可能使用相同的代号。每种寿司的份数都是无限的,Kiana也可以无限次取寿司来吃,但每种寿司每次只能......
  • 【题解】[JLOI2014]镜面通道
    题目描述:在一个二维平面上,有一个镜面通道,由镜面\(AC,BD\)组成,\(AC,BD\)长度相等,且都平行于\(x\)轴,\(B\)位于\((0,0)\)。通道中有\(n\)个外表面为镜面的光学元件,光学元件\(\alpha\)为圆形,光学元件\(\beta\)为矩形(这些元件可以与其他元件和通道有交集,具体看下图)。......
  • HarmonyOS在SDK9版本下FA模型geolocation无法定位问题解决
    问题描述已经在config.json中加入了ohos.permission.LOCATION权限声明,但是在实际开发中,我使用geolocation.getCurrentLocation().then((result)=>{this.locationInfo=JSON.stringify(result);this.blog.setTitle(this.locationInfo);});获取位置信息得不到结果我使用的......
  • POJ2117 Electricity 题解 tarjan点双连通分量 割点
    题目链接:http://poj.org/problem?id=2117题目大意:给定一个由\(n\)个点\(m\)条边构成的无向图,请你求出该图删除一个点之后,连通块最多有多少。解题思路:tarjan,判断\(u\)的子节点有几个\(v\)满足\(low[v]\gedfn[u]\)就是答案,但是同时如果\(u\)不是这个dfs树的根节......
  • CF1697F 题解
    题意传送门构造一个长度为\(n\)的数列\(a\),满足\(1\lea_i\lek\)且\(a\)不降,以及\(m\)个约束,有三种情况:1ix,表示\(a_i\nex\)2ijx,表示\(a_i+a_j\lex\)3ijx,表示\(a_i+a_j\gex\)无解输出\(-1\)。\(1\len,m\le2\times10^4,2\lek\le10\)。题......
  • 【Android】ListView与Button的共存问题解决
    【Android】ListView与Button的共存问题解决这两天在捣鼓ListViewwidget,为了在ListView中加入Button这类的有“点击”事件的widget,请教了不少高手,感谢LandMark对我的认真讲解,下面把解决过程描述一下。 ListView和其它能触发点击事件的widget无法一起正常工作的......
  • P2860 [USACO06JAN]Redundant Paths G 题解 ratjan边双连通分量
    题目链接:https://www.luogu.com.cn/problem/P2860题目大意:给定一个无向连通图,求至少加几条边,能使其变成一个边双连通图。解题思路:边双连通分量缩点后计算度数为\(1\)的节点个数,假设有\(cnt\)个,则答案为\((cnt+1)/2\)。示例程序:#include<bits/stdc++.h>usingnamespac......
  • Educational Codeforces Round 150 (Rated for Div. 2) 题解
    https://codeforces.com/contest/1841https://codeforces.com/contest/1841/problemsD.PairsofSegmentshttps://codeforces.com/contest/1841/problem/D因为\(n\)只有\(2000\),所以考虑枚举每一对\((i,j)\)满足区间有交集并且\(i\neqj\)。如果有交集,就合并。然后......
  • 题解 P9196【[JOI Open 2016] 销售基因链】
    套路题,来讲个套路解法。如果没有后缀的要求,答案就是trie树的子树内字符串数量。现在加上了后缀,尝试继续使用trie树解决问题。我们建立两棵trie树\(T_1,T_2\),其中\(T_1\)是正常的trie树,\(T_2\)是每个字符串翻转后的trie树。这样的话,包含给定后缀的字符串在\(T_2\)......