zoukankan      html  css  js  c++  java
  • [POI2014]Hotel

    Description
    有一个树形结构的宾馆,n个房间,n-1条无向边,每条边的长度相同,任意两个房间可以相互到达。吉丽要给他的三个妹子各开(一个)房(间)。三个妹子住的房间要互不相同(否则要打起来了),为了让吉丽满意,你需要让三个房间两两距离相同。
    有多少种方案能让吉丽满意?

    Input
    第一行一个数n。
    接下来n-1行,每行两个数x,y,表示x和y之间有一条边相连。

    Output
    让吉丽满意的方案数。

    Sample Input
    7
    1 2
    5 7
    2 5
    2 3
    5 6
    4 5

    Sample Output
    5

    HINT
    n≤5000


    这题(O(n^2))即可,但是我TM开始想了个巨麻烦的做法。。。直接设(f[i][j])表示深度为(i)可以凑出的(j)元组的个数,枚举根之后跑3次dfs即可(加强版需要用长链剖分+dp,我不会)

    /*program from Wolfycz*/
    #include<cmath>
    #include<cstdio>
    #include<cstring>
    #include<iostream>
    #include<algorithm>
    #define inf 0x7f7f7f7f
    using namespace std;
    typedef long long ll;
    typedef unsigned int ui;
    typedef unsigned long long ull;
    inline char gc(){
    	static char buf[1000000],*p1=buf,*p2=buf;
    	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
    }
    inline int frd(){
    	int x=0,f=1;char ch=gc();
    	for (;ch<'0'||ch>'9';ch=gc())	if (ch=='-')    f=-1;
    	for (;ch>='0'&&ch<='9';ch=gc())	x=(x<<1)+(x<<3)+ch-'0';
    	return x*f;
    }
    inline int read(){
    	int x=0,f=1;char ch=getchar();
    	for (;ch<'0'||ch>'9';ch=getchar())	if (ch=='-')	f=-1;
    	for (;ch>='0'&&ch<='9';ch=getchar())	x=(x<<1)+(x<<3)+ch-'0';
    	return x*f;
    }
    inline void print(int x){
    	if (x<0)    putchar('-'),x=-x;
    	if (x>9)	print(x/10);
    	putchar(x%10+'0');
    }
    const int N=5e3;
    int pre[(N<<1)+10],now[N+10],child[(N<<1)+10];
    ll f[N+10][4],Ans;
    int n,tot;
    void join(int x,int y){pre[++tot]=now[x],now[x]=tot,child[tot]=y;}
    void insert(int x,int y){join(x,y),join(y,x);}
    void dfs(int x,int fa,int deep,int T){
    	f[deep][T]+=f[deep][T-1];
    	for (int p=now[x],son=child[p];p;p=pre[p],son=child[p]){
    		if (son==fa)	continue;
    		dfs(son,x,deep+1,T);
    	}
    }
    void work(int x){
    	memset(f,0,sizeof(f));
    	for (int i=1;i<=n;i++)	f[i][0]=1;
    	for (int p=now[x],son=child[p];p;p=pre[p],son=child[p])
    		dfs(son,x,1,3),dfs(son,x,1,2),dfs(son,x,1,1);
    	for (int i=1;i<=n;i++)	Ans+=f[i][3];
    }
    int main(){
    	n=read();
    	for (int i=1;i<n;i++){
    		int x=read(),y=read();
    		insert(x,y);
    	}
    	for (int i=1;i<=n;i++)	work(i);
    	printf("%lld
    ",Ans);
    	return 0;
    }
    
  • 相关阅读:
    单元化架构 定义问题
    STGW 下一代互联网标准传输协议QUIC大规模运营之路 wentaomao 腾讯技术工程 2021-02-01
    string
    进程管理工具 源码分析
    etcd 鉴权体系架构由控制面和数据面组成。
    HTTP/2 是基于二进制而不是文本
    分布式 ID 解决方案
    减少重复开发,GraphQL在低代码平台如何落地? 原创 随刻信息流团队 爱奇艺技术产品团队 2021-01-29
    设计模式混编:观察者模式+中介者模式
    mysql 语法总结
  • 原文地址:https://www.cnblogs.com/Wolfycz/p/9949255.html
Copyright © 2011-2022 走看看