【每日一题】7月16日题目精讲—点权和
來源:牛客網:
時間限制:C/C++ 2秒,其他語言4秒 空間限制:C/C++ 131072K,其他語言262144K 64bit IO Format: %lld題目描述
給你一棵樹,最開始點權為0,每次將與一個點x樹上距離<=1的所有點點權+1,之后詢問這些點修改后的點權和.
輸入描述:
第一行兩個數n和m
第二行n-1個數,第i個數fa[i + 1]表示i + 1點的父親編號,保證fa[i + 1]<i + 1
第三行m個數,每個數x依次表示這次操作的點是x
輸出描述:
輸出一個數,即這m次操作的答案的hash值
如果是第i次操作,這次操作結果為ans,則這個hash值加上
i * ans
輸出hash值對19260817取模的結果
示例1
輸入
復制
輸出
復制
示例2
輸入
復制
輸出
復制
備注:
n <= 100000
m <= 10000000
題解:
吐槽一下,題意說的不明確,搞的我一陣子以為自己讀錯了
hash值+=i*ans,而ans等于點權+1后,與點x樹上距離<=1的所有點權和
樣例中最后答案就是:3 * 1+5 * 2+7 * 3=34
根據題意,當影響一個點x后,x本身,x的父親,x的兒子,都會受到影響
int fa[x];//父親節點
int now[x],son[x],grandson[x];//分別表示當前點被操作次數,當前點的孩子節點操作次數,當前點的孫子節點操作次數
int kid[x];//x的兒子節點有多少個
為啥要用當前點的兒子和孫子節點,我是這么理解的,因為每個點的父親節點是唯一的,一旦確定點x,其父親節點,祖父節點都可以確定,而一個點有未知個兒子節點,所以用kid來記錄兒子節點的數量。
最后計算x的貢獻其實是x本身,x的父親,x的兒子三個貢獻和:
x的父親:now[fa[x]]+son[fa[x]]+now[fa[fa[x]]]
x的父親節點的點權受x的父親節點本身,x節點以及x的父親的父親節點
x本身:now[fa[x]]+now[x]+son[x]
x的孩子:kid[x]*now[x]+son[x]+grandson[x]
代碼:
代碼只過了70%,我也沒找到哪里錯了,愁。。
#include<iostream> #include<cstdio> #include<cmath> #include<algorithm> #include<cstring> #define maxn 10000008 const int mod=19260817; typedef long long ll; using namespace std; int fa[maxn]; int now[maxn],son[maxn],grandson[maxn]; int kid[maxn]; int main(){int n,m;cin>>n>>m;for(int i=2;i<=n;i++){int x;cin>>x;fa[i]=x;kid[x]++;}ll sum=0;for(int i=1;i<=m;i++){int x;cin>>x;now[x]=(now[x]+1)%mod;if(fa[x])son[fa[x]]++;if(fa[fa[x]])grandson[fa[fa[x]]]++;int f=(now[fa[x]]+son[fa[x]]+now[fa[fa[x]]])%mod;int m=(now[fa[x]]+now[x]+son[x])%mod;int s=(kid[x]*now[x]%mod+son[x]+grandson[x])%mod;ll ans=(f+m+s)%mod;sum=(sum+ans*i)%mod;} cout<<sum%mod;return 0; }總結
以上是生活随笔為你收集整理的【每日一题】7月16日题目精讲—点权和的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 机构报告 AMD 2023Q3 业绩:在
- 下一篇: 2020牛客暑期多校训练营(第四场)