这道题是2019CCPC河南省省赛的H题,当时没做出来,赛后听学长讲了讲思路,发现其实也不难。比赛的时候写了一个假算法。感觉没有问题,我就写到后面了。如果有大佬能够指出错误,感激不尽。
题目链接:http://acm.zzuli.edu.cn/problem.php?id=2525
题目大意:中文题。
思路:按照给出的答案序列对树上的节点进行优先级划分。然后DFS遍历的时候优先遍历高优先级的(出现位置靠前的)。然后得出DFS序列后和Ans比较即可。
ACCode:
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<map>
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
#define Pair pair<int,int>
#define M_P(a,b) make_pair(a,b)
//#define max(a,b) (a)>(b)?(a):(b)
//#define min(a,b) (a)<(b)?(a):(b)
#define clean(a,b) memset(a,b,sizeof(a))// ??
//std::ios::sync_with_stdio(false);
// register
const int MAXN=1e6+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
vector<int> Vec[MAXN];
int Pre[MAXN],Vis[MAXN];
int Ans[MAXN],tot;
int a[MAXN],b[MAXN];
int n,m;
void Intt(){
for(int i=1;i<=n;++i){
Vec[i].clear();
Pre[i]=-1;Vis[i]=0;
}tot=0;
}
void DFS(int u,int fa){
int len=Vec[u].size();
for(int i=0;i<len;++i){
DFS(Vec[u][i],u);
}Ans[++tot]=u;
}
void Mark(int x){
while(1){
if(Pre[x]==-1||Vis[x]) break;
Vis[x]=1;
Vec[Pre[x]].push_back(x);
x=Pre[x];
}
}
int main(){
int T;scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
Intt();
for(int i=1;i<n;++i){
scanf("%d",&a[i]);
Pre[i+1]=a[i];
}
for(int i=1;i<=m;++i) scanf("%d",&b[i]);
for(int i=1;i<=m;++i) Mark(b[i]);
DFS(1,-1);
int i,j;
for(i=1,j=1;i<=m&&j<=tot;++j){
if(b[i]==Ans[j]){
++i;
}
}//cout<<i<<" "<<j<<endl;
if(i<=m) printf("BAD GUGU\n");
else printf("NOT BAD\n");
}
}
下面的就是我当时想出的假算法:
思路是,用DFS序处理出节点子树的范围,然后每当读入一个节点的时候,将它所在的子树都标记(砍掉)。然后读入节点的时候判断该节点是否在被砍掉的子树中(DFS序标记的区间)。如果在被砍掉的区间内,不符合要求,反之继续,直到遍历所有的b数组。
Code:
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<map>
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
#define Pair pair<int,int>
#define M_P(a,b) make_pair(a,b)
//#define max(a,b) (a)>(b)?(a):(b)
//#define min(a,b) (a)<(b)?(a):(b)
#define clean(a,b) memset(a,b,sizeof(a))// ??
//std::ios::sync_with_stdio(false);
// register
const int MAXN=1e6+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
struct Node{
int v,val,nxt;
Node(int _v=0,int _val=0,int _nxt=0){
v=_v;val=_val;nxt=_nxt;
}
};
Node Edge[MAXN<<1];
int Head[MAXN],Ecnt;
int L[MAXN],R[MAXN],tot;
int Vis[MAXN];
int a[MAXN],b[MAXN];
int n,m;
void Intt(){
for(int i=1;i<=n+10;++i){
Head[i]=-1;Vis[i]=0;
}tot=0;Ecnt=0;
}
void Add(int u,int v,int val){
Edge[Ecnt]=Node(v,val,Head[u]);
Head[u]=Ecnt++;
}
void DFS(int u,int fa){
L[u]=++tot;
for(int i=Head[u];i+1;i=Edge[i].nxt){
int temp=Edge[i].v;
DFS(temp,u);
}R[u]=tot;
}
int main(){
int T;scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
Intt();
for(int i=1;i<n;++i){
scanf("%d",&a[i]);
Add(a[i],i+1,1);
}DFS(1,-1);
// for(int i=1;i<=n;++i){
// cout<<L[i]<<" "<<R[i]<<endl;
// }
for(int i=1;i<=m;++i){
scanf("%d",&b[i]);
}int flag=1;
for(int i=1;i<=m;++i){
int x=b[i];
if(Vis[L[x]]){//该点被标记了
//L[x]表示的是该节点的起始位置,如果这个位置被砍掉了,后面的必定也被砍掉了
flag=0;break;
}
for(int j=L[x];j<=R[x];++j){
// if(Vis[j]==-1) j=R[j];
Vis[j]=1;
}
}
if(flag==0) printf("BAD GUGU\n");
else printf("NOT BAD\n");
}
}
/*
10 5
1 1 1 2 2 3 4 4 8
8 9 4 6 10
*/