牛客——点权和(思维)

点权和

https://ac.nowcoder.com/acm/problem/14393

牛客——点权和

(4个月前好像见过这个题?)

思路:

考虑每次修改后对答案的贡献,对于一个点来说,考虑这个点会影响到哪些点。

考虑只进行一次更新的情况:首先,就是本身,本身的节点值对答案的贡献是度数+1,因为本身的点权会更新,邻接点也会更新;再者,就是他的父节点,父节点对答案的贡献是2,因为在这个过程中本身和父节点的点权都+1了;最后就是他父节点的父节点,因为父节点的更新使得爷爷节点对答案的贡献为1;再考虑他的父节点的其他子节点对他的贡献,这时候要注意去重。

多次的话,我们只需要维护cnt1[x]表示x的操作次数,cnt2[x]表示x的父节点的子节点(即x的兄弟)的操作次数和res[x]表示答案,然后计算对应贡献即可。

代码:

#pragma GCC optimize(2)
#pragma GCC optimize("Ofast","unroll-loops","omit-frame-pointer","inline")
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
typedef pair<double,double>PDD;
#define I_int ll
#define modl 19260817*19890604-19491001
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
char F[200];
inline void out(I_int x) {
    if (x == 0) return (void) (putchar('0'));
    I_int tmp = x > 0 ? x : -x;
    if (x < 0) putchar('-');
    int cnt = 0;
    while (tmp > 0) {
        F[cnt++] = tmp % 10 + '0';
        tmp /= 10;
    }
    while (cnt > 0) putchar(F[--cnt]);
    //cout<<" ";
}
const int maxn=1e6+7,mod=19260817;

ll n,m,fa[maxn],d[maxn];
ll cnt1[maxn],cnt2[maxn],res[maxn];

int main(){
    n=read(),m=read();
    for(int i=2;i<=n;i++){
        fa[i]=read();
        d[i]++;d[fa[i]]++;
    }
    ll sum=0;
    for(int i=1;i<=m;i++) {
        int x=read();
        res[x]=(res[x]+d[x]+1)%mod;
        if(fa[x]){
            res[fa[x]]=(res[fa[x]]+2)%mod;
            cnt2[fa[x]]=(cnt2[fa[x]]+1)%mod;
        }
        if(fa[fa[x]]){
            res[fa[fa[x]]]=(res[fa[fa[x]]]+1)%mod;
        }
        cnt1[x]=(cnt1[x]+1)%mod;
        ll tmp=0;
        if(fa[x]){
            tmp=(tmp+2*cnt1[fa[x]]%mod)%mod;
            if(tmp<0) tmp+=mod;
            tmp=(tmp+(cnt2[fa[x]]-cnt1[x]))%mod;
            if(tmp<0) tmp+=mod;
        }
        if(fa[fa[x]]) tmp=(tmp+cnt1[fa[fa[x]]])%mod;
        tmp=(tmp+res[x])%mod;
        sum=(sum+i*tmp%mod)%mod;
    }
    out(sum);
    return 0;
}

全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务