QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#417289#8720. BFS 序 0HexiuqiWA 274ms65680kbC++143.7kb2024-05-22 17:09:442024-05-22 17:09:45

Judging History

This is the latest submission verdict.

  • [2024-05-22 17:09:45]
  • Judged
  • Verdict: WA
  • Time: 274ms
  • Memory: 65680kb
  • [2024-05-22 17:09:44]
  • Submitted

answer

# include <bits/stdc++.h>
# define int long long
using namespace std;
const int N = 5e5 + 5;
const int inf = 4e18;
int n, q;
int fa[N];
int dfn[N];
int tot;
int siz[N];
int dep[N];
vector <int> e[N];
int a[N];
vector <int> g[N];

void dfs(int u)
{
	siz[u] = 1;
	dfn[u] = ++tot;
	dep[u] = dep[fa[u]] + 1;
	
	for(int i = 0; i < e[u].size(); i++)
	{
		int v = e[u][i];
		dfs(v);
		siz[u] += siz[v];
	}
}

struct segmentTr
{
	int cnt;
	int sum[N * 4];
	int lazy[N * 4];
	int lson[N * 4];
	int rson[N * 4];
	
	void clear()
	{
		cnt = 1;
		lazy[1] = inf;
		sum[1] = 0;
		lson[1] = 0;
		rson[1] = 0;
	}
	
	int newnode(int &x)
	{
		if(x) return x;
		x = ++cnt;
		lazy[x] = inf;
		lson[x] = rson[x] = 0;
		sum[x] = 0;
		return x;
	}
	
	void update(int i, int l, int r, int c)
	{
		sum[i] = (r - l + 1) * c;
		lazy[i] = c;
	}
	
	void down(int i, int l, int r)
	{
		if(lazy[i] == inf)
		{
			return;
		}
		int mid = l + r >> 1;
		update(newnode(lson[i]), l, mid, lazy[i]);
		update(newnode(rson[i]), mid + 1, r, lazy[i]);
		lazy[i] = inf;
	}
	
	void pushup(int i)
	{
		sum[i] = sum[lson[i]] + sum[rson[i]];
	}
	
	void modify(int i, int l, int r, int L, int R, int cov)
	{
		if(R < l || r < L)
		{
			return;
		}
		if(L <= l && r <= R)
		{
			update(i, l, r, cov);
			return;
		}
		down(i, l, r);
		int mid = l + r >> 1;
		modify(newnode(lson[i]), l, mid, L, R, cov);
		modify(newnode(rson[i]), mid + 1, r, L, R , cov);
		pushup(i);
	}
	
	int query(int i, int l, int r, int x)
	{
		if(l == r)
		{
			return sum[i];
		}
		down(i, l, r);
		int mid = l + r >> 1;
		if(x <= mid) return query(newnode(lson[i]), l, mid, x);
		return query(newnode(rson[i]), mid + 1, r, x);
	} 
}Tr;

//int d[N];

void check()
{
//	Tr.clear();
	Tr.modify(1, 1, n, 1, n, 0);
	int m;
	cin >> m;
	map <int, bool> v;
	v.clear();
	for(int i = 1; i <= m; i++)
	{
		cin >> a[i];
	}
	
	for(int i = 1; i <= m; i++)
	{
		if(a[i] < 1|| a[i] > n) 
		{
			cout << "No" << endl;
			return;
		}
	}
	
	for(int i = 1; i <= m; i++)
	{
		if(v[a[i]])
		{
			cout << "No" << endl;
			return;
		}
		v[a[i]] = 1;
	}
	
	for(int i = 1; i <= m; i++)
	{
		if(i > 1 && dep[a[i]] < dep[a[i - 1]])
		{
			cout << "No" << endl;
			return;
		}
	}
	
	for(int i = 1; i <= m; i++)
	{
		g[dep[a[i]]].push_back(a[i]);
	}
	
	bool flag = 1;
	int w = 0;
	
	for(int i = 1; i <= n; i++)
	{
		int precov = 0;
		for(int j = 0; j < g[i].size(); j++)
		{
			int now = g[i][j];
			int cov = Tr.query(1, 1, n, dfn[now]);
			if(precov && cov && dep[a[precov]] == dep[a[cov]])
			{
				if(precov > cov)
				{
					flag = 0;
				}
				
			}
//			cout<<precov<<" "<<now<<" "<<cov<<" "<<i<<endl;
			if(flag == 0) break;
			if(cov) precov = cov;
//			cout<<"*"<<precov<<" "<<cov<<endl; 
			
		}
		if(flag == 0) break;
		for(int j = 0; j < g[i].size(); j++)
		{
			++w;
			int now = g[i][j];
			Tr.modify(1, 1, n, dfn[now], dfn[now] + siz[now] - 1, w);
		}
	}
	
	if(flag)
	{
		cout << "Yes" << endl;
	}
	else
	{
		cout << "No" << endl;
	}
		
	for(int i = 1; i <= m; i++)
	{
		g[dep[i]].clear(); 
	}
}

signed main()
{
	cin >> n;
	
	for(int i = 2; i <= n; i++)
	{
		cin >> fa[i];
		e[fa[i]].push_back(i);
	}
//	bool flag = 0;
//	for(int i = 2; i <= n; i++)
//	{
//		if(fa[i] != 1) flag = 1;
//	}
//	if(!flag) 
//	{
//		cout<<1; 
//	}
	dfs(1);
	
//	for(int i = 1; i <= n; i++)
//	{
//		cout << dfn[i] << " " << siz[i]<<endl;
//	}
	
	cin >> q;
	
	while(q--)
	{
		check();
	}
	
	return 0;
}

/*
6
1 1 3 2 4
12
4 3 2 4 5
4 3 2 5 4
4 3 6 2 5
1 4
3 2 4 5
5 2 5 4 6 3
3 1 4 2
3 5 6 3
5 4 5 2 6 1
4 4 3 2 5
4 4 6 2 3
3 3 2 6

*/

詳細信息

Test #1:

score: 100
Accepted
time: 0ms
memory: 38456kb

input:

6
1 1 3 2 4
10
4 3 6 2 5
1 4
3 2 4 5
5 2 5 4 6 3
3 1 4 2
3 5 6 3
5 4 5 2 6 1
4 4 3 2 5
4 4 6 2 3
3 3 2 6

output:

No
Yes
Yes
No
No
No
No
No
No
Yes

result:

ok 10 token(s): yes count is 3, no count is 7

Test #2:

score: -100
Wrong Answer
time: 274ms
memory: 65680kb

input:

300000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1...

output:

No
Yes
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No
No...

result:

wrong answer expected YES, found NO [3rd token]