【点分治】【map】【哈希表】hdu4670 Cube number on a tree

求树上点权积为立方数的路径数。

显然，分解质因数后，若所有的质因子出现的次数都%3==0，则该数是立方数。

于是在模意义下暴力统计即可。

当然，为了不MLE/TLE，我们不能存一个30长度的数组，而要压成一个long long。

存储状态用map即可，貌似哈希表可以随便卡掉……？

手动开栈……当然这样有可能MLE，所以还得改一些BFS……

<法一>map：

#pragma comment(linker, "/STACK:1024000000,1024000000")

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<map>

#include<queue>

using namespace std;

#define MAXN 50001

typedef pair<int,int> Point;

typedef long long ll;

struct Point2

{

    int x;ll y;int z;

    Point2(){}

    Point2(const int &a,const ll &b,const int &c){x=a;y=b;z=c;}

};

#define MOD 50021

queue<Point2>q;

map<ll,int>ma;

int n,m,ans,ev[MAXN][31];

ll pr[MAXN];

int v[MAXN<<1],first[MAXN],next[MAXN<<1],en;

void AddEdge(const int &U,const int &V)

{

    v[++en]=V;

    next[en]=first[U];

    first[U]=en;

}

bool centroid[MAXN];

int size[MAXN];

int calc_sizes(int U,int Fa)

{

    int res=1;

    for(int i=first[U];i;i=next[i])

      if(v[i]!=Fa&&!centroid[v[i]])

        res+=calc_sizes(v[i],U);

    return size[U]=res;

}

Point calc_centroid(int U,int Fa,int nn)

{

    Point res=make_pair(2147483647,-1);

    int sum=1,maxv=0;

    for(int i=first[U];i;i=next[i])

      if(v[i]!=Fa&&!centroid[v[i]])

        {

          res=min(res,calc_centroid(v[i],U,nn));

          maxv=max(maxv,size[v[i]]);

          sum+=size[v[i]];

        }

    maxv=max(maxv,nn-sum);

    res=min(res,make_pair(maxv,U));

    return res;

}

int en2,en3;

ll td[MAXN],ds[MAXN],base[31];

void calc_dis(int U,int Fa)

{

    td[en2]=0;

    for(int j=1;j<=m;++j)

      td[en2]+=ev[U][j]*base[j-1];

    ++ma[td[en2]];

    q.push(Point2(U,td[en2++],Fa));

    while(!q.empty())

      {

          Point2 Node=q.front(); q.pop();

          for(int i=first[Node.x];i;i=next[i])

            if(v[i]!=Node.z&&!centroid[v[i]])

              {

                ll t=Node.y;

                td[en2]=0;

                for(int j=1;j<=m;++j)

                {

                    td[en2]+=((ev[v[i]][j]+t%3)%3)*base[j-1];

                    t/=3;

                }

                ++ma[td[en2]];

                q.push(Point2(v[i],td[en2++],Node.x));

            }

      }

}

int calc_pairs(ll dis[],int En,int S)

{

    int res=0;

    for(int i=0;i<En;++i)

      {

          ll che=0,t=dis[i];

          for(int j=1;j<=m;++j)

            {

                che+=((6-t%3-ev[S][j])%3)*base[j-1];

                t/=3;

            }

          res+=ma[che];

      }

    return res;

}

void solve(int U)

{

    calc_sizes(U,-1);

    int s=calc_centroid(U,-1,size[U]).second;

    centroid[s]=1;

    for(int i=first[s];i;i=next[i])

      if(!centroid[v[i]])

        solve(v[i]);

    en3=0; ds[en3++]=0;

    ma.clear();

    for(int i=first[s];i;i=next[i])

      if(!centroid[v[i]])

        {

          en2=0; calc_dis(v[i],s);

          ans-=calc_pairs(td,en2,s);

          memcpy(ds+en3,td,en2*sizeof(ll)); en3+=en2;

        }

    ans+=calc_pairs(ds,en3,s);

    centroid[s]=0;

}

void init()

{

    memset(first,0,(n+1)*sizeof(int));

    memset(ev,0,sizeof(ev));

    en=ans=0;

}

int main()

{

    ll t; int a,b; base[0]=1;

    for(int i=1;i<=30;++i) base[i]=base[i-1]*3;

    while(scanf("%d%d",&n,&m)!=EOF)

      {

          init();

          for(int i=1;i<=m;++i) scanf("%I64d",&pr[i]);

          for(int i=1;i<=n;++i)

          {

              bool flag=1;

              scanf("%I64d",&t);

              for(int j=1;j<=m;++j)

                {

                  while(t%pr[j]==0&&t)

                    {

                        ++ev[i][j];

                        t/=pr[j];

                    }

                  ev[i][j]%=3;

                  if(ev[i][j]) flag=0;

                }

              ans+=flag;

          }

        for(int i=1;i<n;++i)

          {

              scanf("%d%d",&a,&b);

              AddEdge(a,b);

              AddEdge(b,a);

          }

        solve(1);

        printf("%d\n",ans);

      }

    return 0;

}

Update:<法二>哈希表，新的姿势，超快的说

#pragma comment(linker, "/STACK:1024000000,1024000000")

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<queue>

using namespace std;

#define MAXN 50001

#define MOD 50021

typedef pair<int,int> Point;

typedef long long ll;

struct Point2

{

	int x;ll y;int z;

	Point2(){}

	Point2(const int &a,const ll &b,const int &c){x=a;y=b;z=c;}

};

struct HashTable

{

    ll v[MOD];

    int en,first[MOD],next[MOD];

    HashTable(){en=0;memset(first,-1,sizeof(first));}

    void clear(){en=0;memset(first,-1,sizeof(first));}

    void insert(const ll &V)

      {

        int U=(int)(V%MOD);

        v[en]=V;

        next[en]=first[U];

        first[U]=en++;

      }

    int count(const ll &V)

      {

          int U=(int)(V%MOD),res=0;

          for(int i=first[U];i!=-1;i=next[i])

            if(v[i]==V)

            ++res;

        return res;

      }

}T;

queue<Point2>q;

int n,m,ans,ev[MAXN][31];

ll pr[MAXN];

int v[MAXN<<1],first[MAXN],next[MAXN<<1],en;

void AddEdge(const int &U,const int &V)

{

	v[++en]=V;

	next[en]=first[U];

	first[U]=en;

}

bool centroid[MAXN];

int size[MAXN];

int calc_sizes(int U,int Fa)

{

	int res=1;

	for(int i=first[U];i;i=next[i])

	  if(v[i]!=Fa&&!centroid[v[i]])

		res+=calc_sizes(v[i],U);

	return size[U]=res;

}

Point calc_centroid(int U,int Fa,int nn)

{

	Point res=make_pair(2147483647,-1);

	int sum=1,maxv=0;

	for(int i=first[U];i;i=next[i])

	  if(v[i]!=Fa&&!centroid[v[i]])

	    {

	      res=min(res,calc_centroid(v[i],U,nn));

	      maxv=max(maxv,size[v[i]]);

	      sum+=size[v[i]];

	    }

	maxv=max(maxv,nn-sum);

	res=min(res,make_pair(maxv,U));

	return res;

}

int En,last;

ll dis[MAXN],base[31];

void calc_dis(int U,int Fa)

{

	dis[En]=0;

	for(int j=1;j<=m;++j)

	  dis[En]+=ev[U][j]*base[j-1];

	q.push(Point2(U,dis[En++],Fa));

	while(!q.empty())

	  {

	  	Point2 Node=q.front(); q.pop();

	  	for(int i=first[Node.x];i;i=next[i])

	  	  if(v[i]!=Node.z&&!centroid[v[i]])

	  	    {

	  	      ll t=Node.y;

	      	  dis[En]=0;

	      	  for(int j=1;j<=m;++j)

	        	{

	          	  dis[En]+=((ev[v[i]][j]+t%3)%3)*base[j-1];

	          	  t/=3;

	        	}

	      	  q.push(Point2(v[i],dis[En++],Node.x));

			}

	  }

}

void calc_pairs(int s)

{

	for(int i=last;i<En;++i)

	  {

	  	ll che=0,t=dis[i];

	  	for(int j=1;j<=m;++j)

	  	  {

	  	  	che+=((6-t%3-ev[s][j])%3)*base[j-1];

	  	  	t/=3;

	  	  }

	  	ans+=T.count(che);

	  }

	for(int i=last;i<En;++i) T.insert(dis[i]);

}

void solve(int U)

{

	calc_sizes(U,-1);

	int s=calc_centroid(U,-1,size[U]).second;

	centroid[s]=1;

	for(int i=first[s];i;i=next[i])

	  if(!centroid[v[i]])

	    solve(v[i]);

	En=0; dis[En++]=0; T.insert(0);

	for(int i=first[s];i;i=next[i])

	  if(!centroid[v[i]])

	    {

	      last=En;

		  calc_dis(v[i],s);

	      calc_pairs(s);

	    }

	for(int i=0;i<En;++i) T.first[dis[i]%MOD]=-1;

	T.en=0;

	centroid[s]=0;

}

void init()

{

	memset(first,0,(n+1)*sizeof(int));

	memset(ev,0,sizeof(ev));

	en=ans=0;

}

int main()

{

	ll t; int a,b; base[0]=1;

	for(int i=1;i<=30;++i) base[i]=base[i-1]*3;

	while(scanf("%d%d",&n,&m)!=EOF)

	  {

	  	init();

	  	for(int i=1;i<=m;++i) scanf("%I64d",&pr[i]);

	  	for(int i=1;i<=n;++i)

		  {

		  	bool flag=1;

		  	scanf("%I64d",&t);

		  	for(int j=1;j<=m;++j)

		  	  {

		  	    while(t%pr[j]==0&&t)

		  	      {

		  	      	++ev[i][j];

		  	      	t/=pr[j];

		  	      }

		  	    ev[i][j]%=3;

		  	    if(ev[i][j]) flag=0;

		  	  }

		  	ans+=flag;

		  }

		for(int i=1;i<n;++i)

		  {

		  	scanf("%d%d",&a,&b);

		  	AddEdge(a,b);

		  	AddEdge(b,a);

		  }

		solve(1);

		printf("%d\n",ans);

	  }

	return 0;

}

<法三>哈希表，旧的姿势，比法一快，比法二慢

#pragma comment(linker, "/STACK:1024000000,1024000000")

#include<cstdio>

#include<algorithm>

#include<cstring>

#include<queue>

using namespace std;

#define MAXN 50001

#define MOD 50021

typedef pair<int,int> Point;

typedef long long ll;

struct Point2

{

    int x;ll y;int z;

    Point2(){}

    Point2(const int &a,const ll &b,const int &c){x=a;y=b;z=c;}

};

struct HashTable

{

    ll v[MOD];

    int en,first[MOD],next[MOD];

    HashTable(){en=0;memset(first,-1,sizeof(first));}

    void clear(){en=0;memset(first,-1,sizeof(first));}

    void insert(const ll &V)

      {

        int U=(int)(V%MOD);

        v[en]=V;

        next[en]=first[U];

        first[U]=en++;

      }

    int count(const ll &V)

      {

          int U=(int)(V%MOD),res=0;

          for(int i=first[U];i!=-1;i=next[i])

            if(v[i]==V)

            ++res;

        return res;

      }

}T;

queue<Point2>q;

int n,m,ans,ev[MAXN][31];

ll pr[MAXN];

int v[MAXN<<1],first[MAXN],next[MAXN<<1],en;

void AddEdge(const int &U,const int &V)

{

    v[++en]=V;

    next[en]=first[U];

    first[U]=en;

}

bool centroid[MAXN];

int size[MAXN];

int calc_sizes(int U,int Fa)

{

    int res=1;

    for(int i=first[U];i;i=next[i])

      if(v[i]!=Fa&&!centroid[v[i]])

        res+=calc_sizes(v[i],U);

    return size[U]=res;

}

Point calc_centroid(int U,int Fa,int nn)

{

    Point res=make_pair(2147483647,-1);

    int sum=1,maxv=0;

    for(int i=first[U];i;i=next[i])

      if(v[i]!=Fa&&!centroid[v[i]])

        {

          res=min(res,calc_centroid(v[i],U,nn));

          maxv=max(maxv,size[v[i]]);

          sum+=size[v[i]];

        }

    maxv=max(maxv,nn-sum);

    res=min(res,make_pair(maxv,U));

    return res;

}

int en2,en3;

ll td[MAXN],ds[MAXN],base[31];

void calc_dis(int U,int Fa)

{

    td[en2]=0;

    for(int j=1;j<=m;++j)

      td[en2]+=ev[U][j]*base[j-1];

    T.insert(td[en2]);

    q.push(Point2(U,td[en2++],Fa));

    while(!q.empty())

      {

          Point2 Node=q.front(); q.pop();

          for(int i=first[Node.x];i;i=next[i])

            if(v[i]!=Node.z&&!centroid[v[i]])

              {

                ll t=Node.y;

                td[en2]=0;

                for(int j=1;j<=m;++j)

                {

                    td[en2]+=((ev[v[i]][j]+t%3)%3)*base[j-1];

                    t/=3;

                }

                T.insert(td[en2]);

                q.push(Point2(v[i],td[en2++],Node.x));

            }

      }

}

int calc_pairs(ll dis[],int En,int S)

{

    int res=0;

    for(int i=0;i<En;++i)

      {

          ll che=0,t=dis[i];

          for(int j=1;j<=m;++j)

            {

                che+=((6-t%3-ev[S][j])%3)*base[j-1];

                t/=3;

            }

          res+=T.count(che);

      }

    return res;

}

void solve(int U)

{

    calc_sizes(U,-1);

    int s=calc_centroid(U,-1,size[U]).second;

    centroid[s]=1;

    for(int i=first[s];i;i=next[i])

      if(!centroid[v[i]])

        solve(v[i]);

    en3=0; ds[en3++]=0;

    T.clear();

    for(int i=first[s];i;i=next[i])

      if(!centroid[v[i]])

        {

          en2=0; calc_dis(v[i],s);

          ans-=calc_pairs(td,en2,s);

          memcpy(ds+en3,td,en2*sizeof(ll)); en3+=en2;

        }

    ans+=calc_pairs(ds,en3,s);

    centroid[s]=0;

}

void init()

{

    memset(first,0,(n+1)*sizeof(int));

    memset(ev,0,sizeof(ev));

    en=ans=0;

}

int main()

{

    ll t; int a,b; base[0]=1;

    for(int i=1;i<=30;++i) base[i]=base[i-1]*3;

    while(scanf("%d%d",&n,&m)!=EOF)

      {

          init();

          for(int i=1;i<=m;++i) scanf("%I64d",&pr[i]);

          for(int i=1;i<=n;++i)

          {

              bool flag=1;

              scanf("%I64d",&t);

              for(int j=1;j<=m;++j)

                {

                  while(t%pr[j]==0&&t)

                    {

                        ++ev[i][j];

                        t/=pr[j];

                    }

                  ev[i][j]%=3;

                  if(ev[i][j]) flag=0;

                }

              ans+=flag;

          }

        for(int i=1;i<n;++i)

          {

              scanf("%d%d",&a,&b);

              AddEdge(a,b);

              AddEdge(b,a);

          }

        solve(1);

        printf("%d\n",ans);

      }

    return 0;

}

【点分治】【map】【哈希表】hdu4670 Cube number on a tree的更多相关文章

HDU4670 cube number on a tree(点分治+三进制加法)
The country Tom living in is famous for traveling. Every year, many tourists from all over the world ...
HDU4670 Cube number on a tree 树分治
人生的第一道树分治,要是早点学我南京赛就不用那么挫了,树分治的思路其实很简单,就是对子树找到一个重心(Centroid),实现重心分解,然后递归的解决分开后的树的子问题,关键是合并,当要合并跨过重心的 ...
[hdu4670 Cube number on a tree]点分治
题意:给一个N个带权节点的树,权值以给定的K个素数为因子,求路径上节点乘积为立方数的路径条数思路:立方数的性质是每个因子的个数为3的倍数,那么每个因子只需要保存0-2三个状态即可,然后路径就可以转化 ...
3.10 Go Map哈希表
3.10 Go Map哈希表 map是key-value类型数据结构,读作(哈希表.字典),是一堆未排序的键值对集合. map是引用类型,使用make函数或者初始化表达式创建. map的key必须是支 ...
hdu 4670 Cube number on a tree（点分治）
Cube number on a tree Time Limit: 20000/10000 MS (Java/Others) Memory Limit: 65535/65535 K (Java/ ...
HDU 4670 Cube number on a tree ( 树的点分治 )
题意 : 给你一棵树 . 树的每一个结点都有一个权值 . 问你有多少条路径权值的乘积是一个全然立方数 . 题目中给了你 K 个素数 ( K <= 30 ) , 全部权值都能分解成这k个素数思路 ...
HDU 4670 Cube number on a tree
divide and conquer on tree. #include <map> #include <vector> #include <cstdio> #in ...
memcached哈希表操作主要逻辑笔记
以下注释的源代码都在memcached项目的assoc.c文件中 /* how many powers of 2's worth of buckets we use */ unsigned int h ...
[CareerCup] 13.2 Compare Hash Table and STL Map 比较哈希表和Map
13.2 Compare and contrast a hash table and an STL map. How is a hash table implemented? If the numbe ...

随机推荐

2016广东工业大学校赛 D题 GDUT-oj1172
Problem D: 二叉树的中序遍历 Description 对于学过数据结构的人来说,二叉树和二叉树上的中序遍历都是再简单不过的东西了.这道题就搞搞二叉树好了,当然,不是一般的二叉树:) 我们定义 ...
Welcome to ShangHai <码农上漂记>
来上海这边快三周了,一切都还算顺利,多亏了朋友们的帮助,要不就得街头打地铺了.对于上海这样的大都市,年轻的我们都想挤一挤,凑凑热闹,实现自己的小小抱负.然而不是每个人都混的起的.以前的我还总想着来大城 ...
c++(类) this指针
this指针的相关概念: this只能在成员函数中使用.全局函数,静态函数都不能使用this.实际上,成员函数默认第一个参数为T* const register this. 为什么this指针不能再静 ...
【bzoj4310/hdu5030-跳蚤】后缀数组
我真的是..调了一百年.. 傻逼的人生.. 而且这题好像可以用sam做哎!我Y出了一个奇怪的办法.. 好吧sam是不能做这题的.搞错了. 说说后缀数组好了.. 搞后缀数组然后我们要二分一个子串,判断 ...
【洛谷 P3805】【模板】manacher算法
题目链接 manacher算法:在线性时间内求一个字符串中所有/最长回文串的算法. 先来考虑一下暴力的算法,枚举每个中点,向两边扩展,时间复杂度\(O(n^2)\). 来分析下此算法的缺点. 1.因为 ...
bzoj 2819 博弈论
我们可以把 n为偶数的时候,n*n的棋盘看成若干个不相交的2*1的格子,那么对于每个2*1的格子,如果先手选了其中的一个,另一个人都可以选另一个,所以最后使先手没有可以选的格子,先手必败(这里的先手并 ...
nginx中fastcgi_params配置参数
Nginx 的 fastcgi 模块提供了 fastcgi_param 指令来主要处理这些映射关系,下面 Ubuntu 下 Nginx 的一个配置文件,其主要完成的工作是将 Nginx 中的变量翻译成 ...
时间模块(time/date)
在Python中,常用的表示方式的时间有:时间戳,字符串时间,元组时间(既年,月,日,时,分,秒,周几,一年中的第几天,时区) time模块: time.timezone: 获取当前标准时 ...
concurrent
from concurrent.futures import ThreadPoolExecutor
RabbitMQ消息队列（二）: 工作队列
1. 工作队列: 对于资源密集型任务,我们等待其处理完成在很多情况下是不现实的,比如无法在http的短暂请求窗口中处理大量耗时任务, 为了达到主线程无需等待,任务异步执行的要求,我们可以将任务加入任务 ...

【点分治】【map】【哈希表】hdu4670 Cube number on a tree

【点分治】【map】【哈希表】hdu4670 Cube number on a tree的更多相关文章

随机推荐

热门专题