The Preliminary Contest for ICPC Asia Shenyang 2019： D. Fish eating fruit（树形dp || 点分治）：

于 2020-08-24 20:58:55 发布
阅读量289 收藏 1
点赞数
分类专栏： dp一般看规律 树形dp 分治/点分治/边分治
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/zhaoxinxin1234/article/details/100834573
版权

真的是诸事不顺，先被k题卡了题意。真水题被卡题意可还行。。。
然后，，就没有然后了。
上手这个题的时候还有半个小时，看起来挺水的一个树形dp，两遍dfs搞定。
可能是因为当时感觉时间不够了，有点紧张，在第一遍dfs的时候忘了加子节点的距离和。
加上就过了。
题目链接：点击这里

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<set>
#include<deque>
#include<map>
#include<vector>
#include<cmath>
#define ll long long
#define llu unsigned ll
#define pr pair<int,int>
using namespace std;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const int inf = 0x3f3f3f3f;
const int maxn=20100;
const int mod=1e9+7;
int head[maxn],ver[maxn],nt[maxn];
ll edge[maxn];
ll dp1[maxn][3],dp2[maxn][3];
ll cnt1[maxn][3],cnt2[maxn][3];
int tot=1;

void add(int x,int y,ll z)
{
    ver[++tot]=y,edge[tot]=z;
    nt[tot]=head[x],head[x]=tot;
}

void dfs1(int x,int fa)
{
    cnt1[x][0]=1;
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(y==fa) continue;
        dfs1(y,x);
        for(int j=0;j<3;j++)
        {
            cnt1[x][(j+z)%3]=(cnt1[x][(j+z)%3]+cnt1[y][j])%mod;
            dp1[x][(j+z)%3]=(dp1[x][(j+z)%3]+dp1[y][j]+z*(cnt1[y][j]))%mod;
        }
    }
}

void dfs2(int x,int fa)
{
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(y==fa) continue;
        for(int j=0;j<3;j++)
        {
            cnt2[y][(j+z)%3]=((cnt1[x][j]-cnt1[y][((j-z)%3+3)%3]+cnt2[x][j])%mod+mod)%mod;
            dp2[y][(j+z)%3]=(((dp1[x][j]-dp1[y][((j-z)%3+3)%3]-z*cnt1[y][((j-z)%3+3)%3])%mod+mod)%mod+dp2[x][j]+z*cnt2[y][(z+j)%3])%mod;
        }
		dfs2(y,x);
    }
}
int main(void)
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(head,0,sizeof(head));
        memset(dp1,0,sizeof(dp1));
        memset(dp2,0,sizeof(dp2));
        memset(cnt1,0,sizeof(cnt1));
        memset(cnt2,0,sizeof(cnt2));
        tot=1;
        int x,y;
        ll z;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%lld",&x,&y,&z);
            x++,y++;
            add(x,y,z);
            add(y,x,z);
        }

        dfs1(1,0);
        dfs2(1,0);

        ll ans[3];
        ans[0]=ans[1]=ans[2]=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<3;j++)
                ans[j]=(ans[j]+dp1[i][j]+dp2[i][j])%mod;
        }

        printf("%lld %lld %lld\n",ans[0],ans[1],ans[2]);
    }
    return 0;
}

现在想一下，其实点分治也可以写，当时就觉得这个题的数据量挺像点分治的。
不过后来被否了。于是写了dp却没有过。话说一上来就点分的话说不定就过了。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;
const int maxn=21000;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
int head[maxn],ver[maxn],nt[maxn];
ll edge[maxn];
int max_part[maxn],_size[maxn];
bool ha[maxn];
int tot=1,root,max_size,n;
ll ans0,ans1,ans2;
ll cnt[3],dis[3],ans[3];

void init(void)
{
    memset(head,0,sizeof(head));
    memset(ha,0,sizeof(ha));
    tot=1,ans0=ans1=ans2=0;
}

void add(int x,int y,ll z)
{
    ver[++tot]=y,edge[tot]=z;
    nt[tot]=head[x],head[x]=tot;
}

void dfs_size(int x,int fa)
{
    _size[x]=1,max_part[x]=0;
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        if(ha[y]||y==fa) continue;
        dfs_size(y,x);
        _size[x]+=_size[y];
        max_part[x]=max(max_part[x],_size[y]);
    }
}

void dfs_root(int now_root,int x,int fa)
{
    max_part[x]=max(max_part[x],_size[now_root]-_size[x]);

    if(max_size>max_part[x])
    {
        max_size=max_part[x];
        root=x;
    }
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        if(ha[y]||y==fa) continue;
        dfs_root(now_root,y,x);
    }
}

void dfs_dis(int x,int fa,ll diss)
{
    cnt[diss%3]++,cnt[diss%3]%=mod;
    dis[diss%3]+=diss,dis[diss%3]%=mod;


    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(ha[y]||y==fa) continue;
        dfs_dis(y,x,diss+z);
    }
}

void get_num(int x,ll d)
{

    dfs_dis(x,0,d);
    ans[0]=ans[1]=ans[2]=0;
    cnt[0]=cnt[1]=cnt[2]=0;
    dis[0]=dis[1]=dis[2]=0;
    dfs_dis(x,0,d);
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
            ans[(i+j)%3]=(ans[(i+j)%3]+cnt[i]*dis[j]+cnt[j]*dis[i])%mod;
    }
}

void dfs(int x)
{
    max_size=inf;
    dfs_size(x,-1);
    dfs_root(x,x,-1);
    get_num(root,0);
    ans0=(ans0+ans[0])%mod;
    ans1=(ans1+ans[1])%mod;
    ans2=(ans2+ans[2])%mod;
    ha[root]=1;
    for(int i=head[root];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(ha[y]) continue;
        get_num(y,z);

        ans0=((ans0-ans[0])%mod+mod)%mod;
        ans1=((ans1-ans[1])%mod+mod)%mod;
        ans2=((ans2-ans[2])%mod+mod)%mod;

        dfs(y);
    }
}

int main(void)
{
    while(scanf("%d",&n)!=EOF)
    {
        init();
        int x,y;
        ll z;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%lld",&x,&y,&z);
            x++,y++;
            add(x,y,z);add(y,x,z);
        }
        dfs(1);

        printf("%lld %lld %lld\n",ans0,ans1,ans2);
    }
    return 0;
}

不过网上大多数的点分治代码，求重心的时候就用了一遍dfs。实际上效率差不多。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<map>
#include<vector>
#define ll long long
#define llu unsigned ll
using namespace std;
const int maxn=21000;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
int head[maxn],ver[maxn],nt[maxn],_size[maxn];;
ll edge[maxn];
bool ha[maxn];
int tot=1,root,max_size,n,si;
ll ans0,ans1,ans2;
ll cnt[3],dis[3],ans[3];

void init(void)
{
    memset(head,0,sizeof(head));
    memset(ha,0,sizeof(ha));
    tot=1,ans0=ans1=ans2=0;
}

void add(int x,int y,ll z)
{
    ver[++tot]=y,edge[tot]=z;
    nt[tot]=head[x],head[x]=tot;
}

void dfs_root(int x,int fa)
{
    _size[x]=1;
    int max_part=0;
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        if(ha[y]||y==fa) continue;
        dfs_root(y,x);
        _size[x]+=_size[y];
        max_part=max(max_part,_size[y]);
    }
    max_part=max(max_part,si-_size[x]);
    if(max_part<max_size) max_size=max_part,root=x;
}

void dfs_dis(int x,int fa,ll diss)
{
    cnt[diss%3]++,cnt[diss%3]%=mod;
    dis[diss%3]+=diss,dis[diss%3]%=mod;


    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(ha[y]||y==fa) continue;
        dfs_dis(y,x,diss+z);
    }
}

void get_num(int x,ll d)
{

    dfs_dis(x,0,d);
    ans[0]=ans[1]=ans[2]=0;
    cnt[0]=cnt[1]=cnt[2]=0;
    dis[0]=dis[1]=dis[2]=0;
    dfs_dis(x,0,d);
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
            ans[(i+j)%3]=(ans[(i+j)%3]+cnt[i]*dis[j]+cnt[j]*dis[i])%mod;
    }
}

void dfs(int x)
{

    get_num(x,0);
    ans0=(ans0+ans[0])%mod;
    ans1=(ans1+ans[1])%mod;
    ans2=(ans2+ans[2])%mod;
    ha[x]=1;
    int totsi=si;
    for(int i=head[x];i;i=nt[i])
    {
        int y=ver[i];
        ll z=edge[i];
        if(ha[y]) continue;
        get_num(y,z);
        ans0=((ans0-ans[0])%mod+mod)%mod;
        ans1=((ans1-ans[1])%mod+mod)%mod;
        ans2=((ans2-ans[2])%mod+mod)%mod;

        max_size=inf;
        si=_size[y]>_size[x]?totsi-_size[x]:_size[y];
        dfs_root(y,0);
        dfs(root);
    }
}

int main(void)
{
    while(scanf("%d",&n)!=EOF)
    {
        init();
        int x,y;
        ll z;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%lld",&x,&y,&z);
            x++,y++;
            add(x,y,z);add(y,x,z);
        }

        si=n;
        max_size=inf;
        dfs_root(1,0);
        dfs(root);

        printf("%lld %lld %lld\n",ans0,ans1,ans2);
    }
    return 0;
}
_Shmily
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
The child health questionnaire: Preliminary data
06-29
The child health questionnaire: Preliminary data T H E CHILD HEALTH QUESTIONNAIRE: PRELIMINARY DATA ALAN C. BUTLER University of Maine In recent years attention has slowly shifted toward an ...
MICROENDOSCOPIC SPECTRAL IMAGING AS A TOOL FOR SMALL DUCTUAL DIAGNOSTICS: PRELIMINARY EXPERIENCE
02-22
We also present the operation principle as well as the preliminary experimental results of a new type of microsize multicore fiber that enables imaging through blood vessel phantoms. Imaging of a ...
The Preliminary Contest for ICPC Asia Shenyang 2019
张宜强的博客
09-14 457
C. Dawn-K's water （完全背包） 样例输入 3 3 2 1 3 1 1 1 3 5 2 3 1 2 3 3 样例输出 3 3 3 6 题意分析： 不同的水有不同的价格，要求买到至少m千克的水，输出花的最少的钱和实际买到的水的重量。 解题思路： 题目和完全背包差不多，主要有两个难点：可以买超过m的水，需要求实际买到的水的重量。 用dpv[i]...
The Preliminary Contest for ICPC Asia Shenyang 2019 G. Special necklace（物理求并联电阻阻值）
lgz0921的博客
09-14 410
题意：求并联电阻的阻值，水题妈蛋的！！！我用java大整数T的要命！，用String就对了！！我服了！！ 题解：很水，画个图看一下就好了，注意每条边的阻值为3a，为什吗呢，看图： 好啦呀！接下来是推图啦： n=1的时候： n=2的时候： n=3的时候： 是不是很有规律，就像那种除法模拟一样，从下往上搞就完事，因为n很大所以肯定是在并联很多之后就没影...
The Preliminary Contest for ICPC Asia Shenyang 2019 C Dawn-K's water （完全背包）
XYQ BLOG
09-14 259
完全背包为什么要取到M，可以取到2*M嘛，这题需要整取，对于不能整取的背包容量，dp[k]=INF，以及dp[j-water[i].weight]=INF时，dp[j]也不需要更新。如果不整取的话，后面无法得知背包是否装满了，只知道，背包容量为m时，所花的最小消耗是多少，但是此时背包可能没装满。 #include <bits/stdc++.h> using namespace std;...
The Preliminary Contest for ICPC Asia Shenyang 2019 D. Fish eating fruit 树形DP 换根与二次扫描
M_theory004的博客
09-17 152
题源：https://nanti.jisuanke.com/t/41403 代码抄借鉴于：https://blog.csdn.net/weixin_44282912/article/details/100833858 #include<iostream> #include<vector> #include<algorithm> #include<st...
2019ICPC(沈阳) - Fish eating fruit(树形dp+树根转移)
Falcon的博客
09-17 297
题目链接：点击查看 题目大意：给出一棵树，给出分类规则：设两点之间的距离为w，则按照w对3取模后的余数分为三组(0,1,2)，问两两顶点之间的距离分类后之和为多少，答案输出模分别为0,1,2时的距离和 题目分析：这个题目需要我们求以每个点为根节点跑一遍dfs然后对距离加和，但显然复杂度为n*n，会T，既然需要每个点都当一次根节点，我们可以试试用树根转移的思想，这样配合上动态规划，就能将时间复杂...
Multivariate regression for pivot selection: A preliminary study
02-08
Pivot selection partially dominates the query performance of pivot-based metric space indexing methods. Lacking of coordinate system in metric spaces, mathematical tools for Rn had been prevented from...
Concurrent and discriminant validity of the Gordon Diagnostic System: A preliminary study
06-29
Concurrent and discriminant validity of the Gordon Diagnostic System: A preliminary study CONCURRENT AND DISCRIMINANT VALIDITY OF THE GORDON DIAGNOSTIC SYSTEM: A PRELIMINARY STUDY JEFFREY N . ...
Preliminary communication: Examination of the harmful effect to fetuses of fumonisin B1 in pregnant sows
06-29
Preliminary communication: Examination of the harmful effect to fetuses of fumonisin B1 in pregnant sows Teratogenesis, Carcinogenesis, and Mutagenesis 20:293–299 (2000) TCM 277 :copyright: 2000 ...
Fish eating fruit
帅比王的博客
09-24 412
测试地址 题意简述： 树上任意两点之间的路径按照模 3 为 012 分类，将两点间距离加和，乘 2 即为答案。 解题思路： 可以采用树上dp解决，也可以点分治，这里先给出一种树上dp做法： dp[i][k] 表示距 i 模 3 为 k 的节点距离和。 tc[i][k] 表示距 i 模 3 为 k 的节点数目。 ans[k] 表示所有路径中模 3 为 k 的路径的总长度。 目标答案是ans[k] ...
# 亚洲区预赛（沈阳）网络赛 D . Fish eating fruit （点分治）
OlaMins
09-16 311
State Z is a underwater kingdom of the Atlantic Ocean. This country is amazing. There are nnn cities in the country and n−1n-1n−1 undirected underwater roads which connect all cities. In order to sav...
The Preliminary Contest for ICPC Asia Shenyang 2019 C. Dawn-K's water
天涯沧海的博客
09-14 241
Input The input consists of multiple test cases, each test case starts with a number nnn (1≤n≤1031 \le n \le 10^31≤n≤103) -- the number of types, and mmm (1≤m≤1041 \le m \le 10^41≤m≤104) -- the least...
「2019沈阳网络赛D」Fish eating fruit【树形dp或树分治】
wzw
09-16 253
题意 就是分别统计所有有向路径长度%3\%3%3意义下的总权值 题解 显然是个换根dpdpdp，需要统计的信息不仅包括模3意义下的总权值，还应包括模3意义下的路径条数，转移起来有点麻烦，建议写树分治 代码 #include<bits/stdc++.h> using namespace std; const int maxn=2e5+10; const long long ...
The Preliminary Contest for ICPC Asia Shenyang 2019 Honk's pool
newhonor
09-14 215
https://nanti.jisuanke.com/t/41406 思路：判断是否都能达到平均值；否则小的一直加k次，大的减k次； 反思：最小的有余数的话不是加1，因为多出的平均加到最小组时，余数使得一些大了，要取最小的，最小的没余数剩下的给； #include <iostream> #include <cstring> #include <queue&...
2019 ICPC 沈阳网络赛 D. Fish eating fruit
蒟蒻的最后的倔强
09-17 272
D. Fish eating fruit 哎 我刚写就想着tarjan 压点 废了 写不出 看题解说了 并查集就开始蛋疼了 菜鸡看到压点就是tarjan 完全忘了并查集了orz n个点，m条边，点分为两种，一种为空房间，一种为怪兽房间。进入空房间可以拿到一个糖果，重复进入不重复计数。进入怪兽房子不能出去，但有一次机会可以 随机 达到一个怪兽房间相邻的房间。询问最多可以找到多少糖果。 必须找最好...
【2019年沈阳网络赛D题】【Fish eating fruit】【树形dp(转移写法)】
流水
09-14 700
题目链接： https://nanti.jisuanke.com/t/41403 题意： 题目给了一个NNN个点N−1N-1N−1条边的树形图 现在每次出行的交通工具是鱼，而鱼对于路径长度有不同的喜好 如果总路径长度是3的倍数，那么鱼需要路径长度数量的 榴莲 如果总路径长度%3=1，那么鱼需要路径长度数量的 木瓜 如果总路径长度%3=2，那么鱼需要路径长度数量的 牛奶果 现在问，从图上的每个...
Fish eating fruit 点分治
weixin_30389003的博客
09-14 98
树上任意两点之间的路径按照模 3 为 012 分类，将两点间距离加和（边权累和） ，乘 2 即为答案。 点分治模板题 #include<iostream> #include<cstdio> #include<algorithm> #include<cstring> using namespace std; ...
CIKM2019：图模型在个性化推荐中的探索与应用
在CIKM2019会议上，一篇名为"Learning and Reasoning on Graph for Recommendation"的论文探讨了如何利用图形模型在推荐系统中的应用。该教程主要分为四个部分： 1. **引言** (Introduction): 开篇首先介绍了推荐...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值