生活随笔
收集整理的這篇文章主要介紹了
P3605 [USACO17JAN]Promotion Counting P dfs序
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
傳送門
文章目錄
題意：
思路：
這個(gè)題可以用各種姿勢(shì)a掉，樹啟和線段樹合并都可以，比較無腦。下面給一個(gè)解決這種問題比較巧妙的做法。
考慮暴力怎么寫，我們先將每個(gè)點(diǎn)的權(quán)值離散化一下，每到一個(gè)點(diǎn)就把樹狀數(shù)組清空，讓后在樹狀數(shù)組中加上這個(gè)子樹的值，用樹狀數(shù)組查詢pj>pip_j>p_ipj?>pi?的數(shù)量。
我們優(yōu)化以上算法，還是先離散化，也是用到了樹狀數(shù)組。
考慮對(duì)于一個(gè)點(diǎn)，當(dāng)dfsdfsdfs遍歷這個(gè)點(diǎn)會(huì)遍歷兩次 ，第一次是從父親節(jié)點(diǎn)到這個(gè)點(diǎn)，第二次是從他的子節(jié)點(diǎn)回溯回來。我們?cè)?strong>回溯的時(shí)候加上這個(gè)節(jié)點(diǎn)的權(quán)值，那么當(dāng)回溯到這個(gè)點(diǎn)的時(shí)候，樹狀數(shù)組中存的值包含了以這個(gè)點(diǎn)為根的子樹的全部權(quán)值，當(dāng)然里面可能還有別的點(diǎn)的值(即"垃圾"值)，如果我們按照之前一遍一清空的話復(fù)雜度是O(n2)O(n^2)O(n2)的，我們就是要優(yōu)化這里?，F(xiàn)在想想，如果能把樹狀數(shù)組中那些”垃圾“值減去不就行了嗎？我們?nèi)绾尾拍苤滥切┲的?#xff1f;我們驚奇的發(fā)現(xiàn)第一次到這個(gè)點(diǎn)的時(shí)候樹狀數(shù)組中存的不就是這個(gè)值嘛？所以我們定義一個(gè)ansansans，第一次到這個(gè)點(diǎn)的時(shí)候減去sum(se)?sum(a[u])sum(se)-sum(a[u])sum(se)?sum(a[u])，回溯的時(shí)候加上sum(se)?sum(a[u])sum(se)-sum(a[u])sum(se)?sum(a[u])即可，這個(gè)時(shí)候即為答案啦。
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
#define lowbit(x) ((x)&(-x))
using namespace std
;
typedef long long LL
;
typedef unsigned long long ULL
;
typedef pair
<int,int> PII
;const int N
=1000010,M
=N
*2,mod
=1e9+7,INF
=0x3f3f3f3f;
const double eps
=1e-6;int n
,se
;
int a
[N
];
vector
<int>v
;
int tr
[N
];
int e
[M
],ne
[M
],h
[N
],idx
;
int ans
[N
];void add(int a
,int b
)
{e
[idx
]=b
,ne
[idx
]=h
[a
],h
[a
]=idx
++;
}void addd(int x
,int c
)
{for(int i
=x
;i
<=se
;i
+=lowbit(i
)) tr
[i
]+=c
;
}int sum(int x
)
{int ans
=0;for(int i
=x
;i
;i
-=lowbit(i
)) ans
+=tr
[i
];return ans
;
}int find(int x
)
{return lower_bound(v
.begin(),v
.end(),x
)-v
.begin();
}void dfs(int u
)
{ans
[u
]-=sum(se
)-sum(a
[u
]);for(int i
=h
[u
];~i
;i
=ne
[i
]) { int ver
=e
[i
]; dfs(ver
); }ans
[u
]+=sum(se
)-sum(a
[u
]);addd(a
[u
],1);
}int main()
{
scanf("%d",&n
);memset(h
,-1,sizeof(h
)); idx
=0;for(int i
=1;i
<=n
;i
++) scanf("%d",&a
[i
]),v
.pb(a
[i
]);sort(v
.begin(),v
.end()); v
.erase(unique(v
.begin(),v
.end()),v
.end());se
=v
.size();for(int i
=1;i
<=n
;i
++) a
[i
]=find(a
[i
])+1;for(int i
=2;i
<=n
;i
++){int x
; scanf("%d",&x
);add(x
,i
);}dfs(1);for(int i
=1;i
<=n
;i
++) printf("%d\n",ans
[i
]);return 0;
}
總結(jié)
以上是生活随笔為你收集整理的P3605 [USACO17JAN]Promotion Counting P dfs序的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。
