zzuli-2525-咕咕的搜索序列(思维+DFS)

这道题是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


*/

猜你喜欢

转载自blog.csdn.net/qq_40482358/article/details/89301346
今日推荐