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=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,m,hsh[MAXN],HSH[MAXN],bl[MAXN],sz[MAXN],dep[MAXN],fa[MAXN];
int nxt[MAXN<<1],fst[MAXN],to[MAXN<<1],cnt;
ll sum[51][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]=1;
    for(int i=fst[x];i;i=nxt[i])
    {
        if(to[i]==fa[x]) continue;
        dep[to[i]]=dep[x]+1,fa[to[i]]=x;
        dfs(to[i]);sz[x]+=sz[to[i]];
    }
}
void DFS(int x,int anc)
{
    int hvs=0;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=1;i<n;i++) {a=read(),b=read();add(a,b);add(b,a);}
    dfs(1);cnt=0;
    DFS(1,1);
    for(int i=1;i<=50;i++)
        for(int j=1;j<=n;j++) (val[j]*=dep[HSH[j]])%=MOD,sum[i][j]=(sum[i][j-1]+val[j])%MOD;
    m=read();
    while(m--)
    {
        a=read(),b=read(),c=read(),res=0;
        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]]-1]+MOD)%=MOD;
            a=fa[bl[a]];
        }
        if(dep[a]>dep[b]) swap(a,b);
        (res+=sum[c][hsh[b]]-sum[c][hsh[a]-1]+MOD)%=MOD;
        printf("%lld
",res);
    }
}