博客
关于我
小 Q 与树(淀粉质)
阅读量:224 次
发布时间:2019-02-28

本文共 2725 字,大约阅读时间需要 9 分钟。

解题思路:点分治。对于每一个重心,找到当前子树的所有符合条件的。

#include
using namespace std;typedef long long ll;typedef long double lf;typedef unsigned long long ull;typedef pair
P;const int inf = 0x7f7f7f7f;const ll INF = 1e16;const int N = 2e5+10;const ull base = 131;const ll mod = 998244353;inline int read(){ int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){ if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){ x=x*10+ch-'0';ch=getchar();}return x*f;}inline string readstring(){ string str;char s=getchar();while(s==' '||s=='\n'||s=='\r'){ s=getchar();}while(s!=' '&&s!='\n'&&s!='\r'){ str+=s;s=getchar();}return str;}int random(int n){ return (int)(rand()*rand())%n;}void writestring(string s){ int n = s.size();for(int i = 0;i < n;i++){ printf("%c",s[i]);}}bool is_prime(int n){ if(n <2) return false;for(int i = 2;i*i <= n;i++){ if(n%i == 0) return false;}return true;}struct str{ int dot; int dis; int next;}e[N<<4];int head[N<<4],C = 0;void add(int u,int v,int z){ e[++C].dis = z; e[C].dot = v; e[C].next = head[u]; head[u] = C;}ll val[N];int dp[N];int sum,rt;int size[N],vis[N];void getrt(int u,int fa){ size[u] = 1;dp[u] = 0; for(int i = head[u];i;i = e[i].next){ int v = e[i].dot; if(v == fa|| vis[v]) continue; getrt(v,u); size[u] += size[v]; dp[u] = max(dp[u],size[v]); } dp[u] = max(dp[u],sum-size[u]); if(dp[u] < dp[rt]) rt = u;}int tot,dis[N];struct node{ ll val; int dis; int dot;}a[N];void getdis(int u,int fa){ a[++tot].val = val[u]; a[tot].dis = dis[u]; a[tot].dot = u; for(int i = head[u];i;i = e[i].next){ int v = e[i].dot; if(v == fa || vis[v]) continue; dis[v] = dis[u]+e[i].dis; getdis(v,u); }}bool cmp(node a,node b){ return a.val < b.val;}ll ans = 0;ll calc(int u,int w){ tot = 0;dis[u] = w; getdis(u,0); sort(a+1,a+1+tot,cmp); ll ret = 0,sum = 0; /* for(int i = 1;i <= tot;i++){ for(int j = i+1;j <= tot;j++){ ret = (ret+a[i].val*(a[i].dis+a[j].dis))%mod; } } */ for(int i = 1;i <= tot;i++){ sum = (sum+a[i].dis)%mod; } for(int i = 1;i < tot;i++){ ll x = a[i].val,y = a[i].dis; sum = (sum-y+mod)%mod; ll len = tot-i; y = (len*x*y)%mod; x = (x*sum)%mod; ret = (ret+x+y)%mod; } ret = (ret+ret)%mod; return ret;}void solve(int u){ vis[u] = 1; //cout<<
<= n;i++){ val[i] = read(); } for(int i = 1;i < n;i++){ int u = read(),v = read(); add(u,v,1); add(v,u,1); } dp[0] = inf;sum = n;rt = 0; getrt(1,0); solve(rt); cout<
<

转载地址:http://rkqp.baihongyu.com/

你可能感兴趣的文章
MySQL清空表数据
查看>>
mysql源码安装
查看>>
Mysql源码安装过程中可能碰到的问题
查看>>
MySQL灵魂16问,你能撑到第几问?
查看>>
MySQL灵魂拷问:36题带你面试通关
查看>>
mysql状态分析之show global status
查看>>
mysql状态查看 QPS/TPS/缓存命中率查看
查看>>
mysql生成树形数据_mysql 实现树形的遍历
查看>>
mysql用于检索的关键字_Mysql全文搜索match...against的用法
查看>>
MySQL用得好好的,为什么要转ES?
查看>>
MySql用户以及权限的管理。
查看>>
MySQL用户权限配置:精细控制和远程访问的艺术!------文章最后有惊喜哦。
查看>>
mysql用户管理、常用语句、数据分备份恢复
查看>>
MySQL留疑问:left join时选on还是where?
查看>>
mysql登陆慢问题解决
查看>>
Mysql百万级数据查询优化
查看>>
MySQL的 DDL和DML和DQL的基本语法
查看>>
mysql的 if else , case when then, IFNULL
查看>>
MySQL的10种常用数据类型
查看>>
MySQL的btree索引和hash索引的区别
查看>>