[BZOJ]3522: [Poi2014]Hotel

　　題解: $ n<=5000 $ 那咱們考慮暴力枚舉根 php

　　$ dp[x] $表示任意兩個屬於不一樣根兒子子樹且深度爲x個數的乘積和ios

　　$ sum[x] $表示不一樣根兒子子樹深度爲x的個數和git

　　而後統計方案就好了spa

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=5e3+10;
const double eps=1e-8;
#define ll long long
using namespace std;
struct edge{int t;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
ll read(){
    ll 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;
}


int n;
ll dp[MAXN],sum[MAXN],num[MAXN],ans;
int maxx;
bool flag;

void dfs(int x,int pre,int deep){
    maxx=max(maxx,deep+1);
    num[deep+1]++;
    if(flag)ans+=dp[deep+1];
    link(x){
	if(j->t==pre)continue;
	dfs(j->t,x,deep+1);
    }
}

int main(){
    n=read();
    int x,y;
    inc(i,2,n)x=read(),y=read(),add(x,y),add(y,x);
    inc(i,1,n){
	int cnt=0;
	link(i){
	    cnt++;
	    if(cnt>=3)flag=1;
	    maxx=0;
	    dfs(j->t,i,1);
	    inc(j,1,maxx)dp[j]+=sum[j]*num[j],sum[j]+=num[j],num[j]=0;
	}
	inc(j,1,n)dp[j]=sum[j]=num[j]=0;
    }
    printf("%lld\n",ans);
}

3522: [Poi2014]Hotel

Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 1250 Solved: 690
[Submit][Status][Discuss]

Description

有一個樹形結構的賓館，n個房間，n-1條無向邊，每條邊的長度相同，任意兩個房間能夠相互到達。吉麗要給他的三個妹子各開（一個）房（間）。三個妹子住的房間要互不相同（不然要打起來了），爲了讓吉麗滿意，你須要讓三個房間兩兩距離相同。
有多少種方案能讓吉麗滿意？blog

Input

第一行一個數n。
接下來n-1行，每行兩個數x,y，表示x和y之間有一條邊相連。ip

Output

讓吉麗滿意的方案數。get

Sample Input