CCPC-Wannafly Winter Camp Day5(Div1, onsite)(Nested Tree-빈 트리)
dp 세트 dp 프리 프로세싱 d p s u m [x] dpsum [x] dpsum [x]는 모든 n개 점의 나무에서 모든 점에서 점 x x까지의 거리와
m 개의 점으로 연결된 새 트리에서 각 복제본과 부모 복제본이 연결된 점은 t o p x topxtopx, 하위 트리 크기 t o t x totxtotx, 서브트리의 모든 점에서 to p x topx topx의 거리 및 d p s x dpsx dpsx. 각 복제본에 대해 모든 포인트 쌍을 집계하는 lca가 해당 복제본의 경로에서 응답에 기여하는 서브트리의 복사본 a n s + = d e p s u m a ∗ t o t v + n ∗ d e p s v + n ∗ t t v ans + = depsuma*tot_v+n*deps_v+n*tot_vans+= depsuma totv +n depsv +n totv 서브트리의 복사{v2}*tot_v+deps_{v}*tot_{v2}+tot_v*tot_{v2}*(2+dis(a,a 2)ans+=depsv2 (topsv2 +depsv totv2 +totv totv2 (2+dis(a,a2)) 중 a와 a2는 각각 서브 트리 v와 서브 트리 v2가 연결된 점을 표시하고 마지막으로 dis(a,a2)dis(a,a2)dis(a,a 2)dis(a,a2)를 계산하기 위해 허수로 처리해야 한다.모든 키 노드를 처리하고 하위 트리 v의 위치 a에 to tv totv totv는 점권, s t o t x stotxstotx는 x의 하위 트리의 점권과all를 가상 트리의 점 및 각 가상 트리의 모서리(x, v)에 대한 기여 a n s + = (d e p v - - d e p x)\s t o t v\(a l l - s t o t v)ans + = (dep v-dep x)*stotv*(all-stot v)ans+= (depv - depx) stotv stotv(all - stotv), 마지막으로 deps를 업데이트하고 허수 위의 변(x, v)이 deps에 기여한 것은(d e p [v] - d e p [x])(a l l -3. s t o t x): s t o t x)(dep[v]-dep[x])* (top x가 v에 있는 하위 트리)?(all-stot_x):stot_x)(dep[v]-3. dep[x])∗(topx가 v에 있는 하위 나무)?(all -3. stotx): stotx) 각 복사 v 대 d e p s deps deps의 공헌은 d e p s v + t o t v depsv+tot_v depsv+totv
#include
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForkD(i,k,n) for(int i=n;i>=k;i--)
#define Rep(i,n) for(int i=0;i
#define ForD(i,n) for(int i=n;i>0;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=next[p])
#define Lson (o<<1)
#define Rson ((o<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,0x3f,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define MEMx(a,b) memset(a,b,sizeof(a));
#define INF (0x3f3f3f3f)
#define F (1000000007)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define vi vector
#define pi pair
#define SI(a) ((a).size())
#define Pr(kcase,ans) printf("Case #%d: %lld
",kcase,ans);
#define PRi(a,n) For(i,n-1) cout<#define PRi2D(a,n,m) For(i,n) { \
For(j,m-1) cout<#pragma comment(linker, "/STACK:102400000,102400000")
#define ALL(x) (x).begin(),(x).end()
#define gmax(a,b) a=max(a,b);
#define gmin(a,b) a=min(a,b);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
ll mul(ll a,ll b){return (a*b)%F;}
ll add(ll a,ll b){return (a+b)%F;}
ll sub(ll a,ll b){return ((a-b)%F+F)%F;}
void upd(ll &a,ll b){a=(a%F+b%F)%F;}
inline int read()
{
int x=0,f=1; char ch=getchar();
while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();}
return x*f;
}
namespace VT{
#define LL long long
int cnt;
int f[300001], g[300001];
unsigned char lg2[1 << 20];
int st[20][600001], len;
LL h[300001];
#ifdef EDGE
vector<pair<int, int>> v[300001];
void dfs(int i, int father)
{
int rank = st[0][len] = ++cnt;
f[cnt] = i; g[i] = len++;
for (unsigned int j = 0; j < v[i].size(); j++){
int k = v[i][j].first;
if (k != father){
h[k] = h[i] + v[i][j].second;
dfs(k, i);
st[0][len++] = rank;
}
}
}
#else
vector<int> v[300001];
void dfs(int i, int father)
{
int rank = st[0][len] = ++cnt;
f[cnt] = i; g[i] = len++;
for (unsigned int j = 0; j < v[i].size(); j++){
int k = v[i][j];
if (k != father){
h[k] = h[i] + 1;
dfs(k, i);
st[0][len++] = rank;
}
}
}
#endif
void init(int root)
{
len = 0; cnt = 0;
#ifdef EDGE
v[0].push_back(make_pair(root, 0));
#else
v[0].push_back(root);
#endif
dfs(0, 0);
for (unsigned char i = 0; (1 << i) <= len; i++)
memset(&lg2[1 << i], i, 1 << i);
for (int i = 1; i <= lg2[len]; i++){
for (int k = len - (1 << i); k >= 0; k--)
st[i][k] = min(st[i - 1][k], st[i - 1][k + (1 << (i - 1))]);
}
}
inline int query(int i, int j){
int pos1 = g[i], pos2 = g[j];
if (pos1 > pos2)swap(pos1, pos2);
int t = lg2[pos2 - pos1];
return f[min(st[t][pos1], st[t][pos2 - (1 << t) + 1])];
}
int node[300001];
int s[300001], top;
vector<int> vt[300001];
bool used[300001];//±£´æÐéÊ÷ÖÐÄÄЩµãÊÇÔÊäÈëµã
inline bool cmp(int i, int j){ return g[i] < g[j]; }
void makeVirtualTree(int n)
{
sort(node, node + n, cmp);
s[top = 0] = 0; node[n] = 0;
for (int i = 0; i <= n; i++){
int t = query(s[top], node[i]);
if (t != s[top]){
while (query(s[--top], t) != s[top]){
vt[s[top]].push_back(s[top + 1]);
}
vt[t].push_back(s[top + 1]);
if (s[top] != t)s[++top] = t;
}
s[++top] = node[i];
used[node[i]] = 1;
}
}
void clearVirtualTree(int i)
{
for (unsigned int j = 0; j < vt[i].size(); j++)
clearVirtualTree(vt[i][j]);
vt[i].clear(); used[i] = 0;
}
}
int n,m;
#define MAXN (112345)
vi e[MAXN];
ll dfn[MAXN]={},totT=0,rr[MAXN]={};
ll dep[MAXN]={},depsum[MAXN]={},depup[MAXN]={},depdown[MAXN]={},sz[MAXN]={};
void dfs(int x,int fa) {
dep[x]=dep[fa]+1;
dfn[x]=++totT;
sz[x]=1;
for(int v:e[x]) if(v^fa) {
dfs(v,x);
sz[x]+=sz[v];
depdown[x]+=depdown[v]+sz[v];
depdown[x]%=F;
}
rr[x]=totT;
}
void dfs2(int x,int fa) {
if(x^1) depup[x]=depup[fa]+depdown[fa]-depdown[x]-sz[x] +n-sz[x];
depup[x]%=F;
for(int v:e[x]) if(v^fa) {
dfs2(v,x);
}
}
struct node{
int b,u,v;
};
vector<node> e2[MAXN];
ll tot[MAXN]={},top[MAXN]={};
ll ans=0;
void dfs3(int x,int fa) {
tot[x]=n;
for(auto pa:e2[x]) {
int v=pa.b,a=pa.u,b=pa.v;
if(v^fa) {
top[v]=b;
dfs3(v,x);
tot[x]=(tot[x]+tot[v])%F;
}
}
}
ll deps[MAXN]={},stot[MAXN]={};
ll all=0,_deps=0;
int _topx=0;
void dfs5(int x,int fa) {
for(auto v:VT::vt[x]) if(v^fa){
dfs5(v,x);
upd(stot[x],stot[v]);
}
}
void dfs6(int x,int fa) {
for(auto v:VT::vt[x]) if(v^fa){
ll w=dep[v]-dep[x];
dfs6(v,x);
upd(ans,mul( mul(stot[v], sub(all,stot[v])), w) );
}
for(auto v:VT::vt[x]) if(v^fa){
ll w=dep[v]-dep[x],p=0;
if(dfn[v]<=dfn[_topx]&& dfn[_topx] <=rr[v] ) p=sub(all,stot[v])*w%F;
else p=stot[v]*w%F;
upd(_deps,p);
}
}
void dfs_clr(int x,int fa) {
stot[x]=0;
for(auto v:VT::vt[x]) if(v^fa){
dfs_clr(v,x);
}
}
void dfs4(int x,int fa) {
ll totv2=0,depsv2=0;
for(auto pa:e2[x]) {
int v=pa.b,a=pa.u,b=pa.v;
if(v^fa) {
dfs4(v,x);
upd(ans,depsum[a]*tot[v]%F+ deps[v]*n%F + n*tot[v]%F );
upd(ans,depsv2*tot[v]%F+deps[v]*totv2%F + tot[v]*totv2%F*2%F);
upd(totv2,tot[v]);
upd(depsv2,deps[v]);
}
}
{
int sn=0;
if(top[x]) VT::node[sn++]=top[x];
for(auto pa:e2[x]) {
int v=pa.b,a=pa.u,b=pa.v;
if(v^fa) {
VT::node[sn++]=a;
}
}
sort(VT::node,VT::node+sn);
sn=unique(VT::node,VT::node+sn)-VT::node;
Rep(i,sn) stot[VT::node[i]]=0;
for(auto pa:e2[x]) {
int v=pa.b,a=pa.u,b=pa.v;
if(v^fa) {
stot[a]+=tot[v];
}
}
VT::makeVirtualTree(sn);
dfs5(0,-1);
all=stot[0]; _topx=top[x]; _deps=0;
dfs6(VT::vt[0][0],-1);
upd(deps[x],_deps);
upd(deps[x],depsum[top[x]]);
for(auto pa:e2[x]) {
int v=pa.b,a=pa.u,b=pa.v;
if(v^fa) {
upd(deps[x],deps[v]+tot[v]);
}
}
dfs_clr(0,-1);
VT::clearVirtualTree(0);
}
}
ll pow2(ll a,int b,ll p) //a^b mod p
{
if (b==0) return 1%p;
if (b==1) return a%p;
ll c=pow2(a,b/2,p)%p;
c=c*c%p;
if (b&1) c=c*a%p;
return c%p;
}
ll inv(ll a,ll p) { //gcd(a,p)=1
return pow2(a,p-2,p);
}
ll inv2=inv(2,F);
int main()
{
// freopen("H.in","r",stdin);
// freopen(".out","w",stdout);
cin>>n>>m;
For(i,n-1) {
int u=read(),v=read();
e[u].pb(v);e[v].pb(u);
VT::v[u].pb(v);
VT::v[v].pb(u);
}
VT::init(1);
dfs(1,0);
dfs2(1,0);
For(x,n) depsum[x]=(depdown[x]+depup[x])%F;
For(i,m-1) {
int a=read(),b=read(),u=read(),v=read();
e2[a].pb(node{b,u,v}), e2[b].pb(node{a,v,u});
}
dfs3(1,0);
dfs4(1,0);
ll ps=0;
For(x,n) upd(ps,depsum[x]*m%F);
ps=ps*inv2%F;
upd(ans,ps);
cout<<ans<<endl;
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
CCPC-Wannafly Winter Camp Day5(Div1, onsite)(Nested Tree-빈 트리)각 복제본에 대해 모든 포인트 쌍을 집계하는 lca가 해당 복제본의 경로에서 응답에 기여하는 서브트리의 복사본 a n s + = d e p s u m a ∗ t o t v + n ∗ d e p s v + n ∗ t ...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.