luogu 4427 求和

bjoi 2018 求和 唯一一道可能切的题一个数组还没开long long就成0分了

题目大意：

一棵有根树，并且希望多次询问这棵树上一段路径上所有节点深度的k次方和，而且每次的k可能是不同的

此处节点深度的定义是这个节点到根的路径上的边数

思路：

考试的时候随便写了个树剖

剖下来之后搞五十个次方的前缀和

然后一个数组乘起来没开long long

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>
 #include<queue>
 #include<vector>
 #define ll long long
 #define MAXN 300100
 #define inf 2139062143
 #define MOD 998244353
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)) {if(ch=='-') f=-;ch=getchar();}
     while(isdigit(ch)) {x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 int n,m,hsh[MAXN],HSH[MAXN],bl[MAXN],sz[MAXN],dep[MAXN],fa[MAXN];
 int nxt[MAXN<<],fst[MAXN],to[MAXN<<],cnt;
 ll sum[][MAXN],val[MAXN],res;
 void add(int u,int v){nxt[++cnt]=fst[u],fst[u]=cnt,to[cnt]=v;}
 void dfs(int x)
 {
     sz[x]=val[x]=;
     for(int i=fst[x];i;i=nxt[i])
     {
         if(to[i]==fa[x]) continue;
         dep[to[i]]=dep[x]+,fa[to[i]]=x;
         dfs(to[i]);sz[x]+=sz[to[i]];
     }
 }
 void DFS(int x,int anc)
 {
     int hvs=;hsh[x]=++cnt,HSH[cnt]=x,bl[x]=anc;
     for(int i=fst[x];i;i=nxt[i])
         if(to[i]!=fa[x]&&sz[to[i]]>sz[hvs])    hvs=to[i];
     if(!hvs) return ;
     DFS(hvs,anc);
     for(int i=fst[x];i;i=nxt[i])
         if(to[i]!=fa[x]&&to[i]!=hvs) DFS(to[i],to[i]);
 }
 int main()
 {
     n=read();int a,b,c;
     for(int i=;i<n;i++) {a=read(),b=read();add(a,b);add(b,a);}
     dfs();cnt=;
     DFS(,);
     for(int i=;i<=;i++)
         for(int j=;j<=n;j++) (val[j]*=dep[HSH[j]])%=MOD,sum[i][j]=(sum[i][j-]+val[j])%MOD;
     m=read();
     while(m--)
     {
         a=read(),b=read(),c=read(),res=;
         while(bl[a]!=bl[b])
         {
             if(dep[bl[a]]<dep[bl[b]]) swap(a,b);
             (res+=sum[c][hsh[a]]-sum[c][hsh[bl[a]]-]+MOD)%=MOD;
             a=fa[bl[a]];
         }
         if(dep[a]>dep[b]) swap(a,b);
         (res+=sum[c][hsh[b]]-sum[c][hsh[a]-]+MOD)%=MOD;
         printf("%lld\n",res);
     }
 }