oj.uz

Submission #953114

# Submission time Handle Problem Language Result Execution time Memory
953114 2024-03-25T14:14:59 Z PM1 File Paths (BOI15_fil) C++17
0 / 100
 20 ms 2796 KB 
#include <bits/stdc++.h>
using namespace std;
const int mxn=3e3+5,mxs=1e6+5;
int n,m,s,k,par[mxn*2],dis[mxn*2];
bitset<mxn>b[mxs];
vector<int>all,v[mxn];
bool mark[mxs],ans[mxn*2];
void make(int x,bool y){
	int z=x;
	while(x!=0){
		x=par[x];
		int d=dis[z]-dis[x];
		if(d<=k)mark[d]=y;
	}
}
void dfs(int z){
	b[z]=((b[par[z]]<<(dis[z]-dis[par[z]]))|b[par[z]]);
	b[z][s]=1;
	for(auto i:v[z]){
		dfs(i);
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m>>k>>s;
	s++;
	for(int i=1;i<=n+m;i++){
		int x,y;
		cin>>x>>y;
		par[i]=x;
		dis[i]=dis[x]+y+1;
		if(i<=n){
			all.push_back(dis[i]);
			v[x].push_back(i);
		}
	}
	all.push_back(0);
	for(int i=n+1;i<=n+m;i++){
		make(i,1);
		if(dis[i]==k)ans[i]=1;
		for(auto j:all){
			int res=k-j-s;
			if(res>=0 && mark[res])ans[i]=1;
		}
		make(i,0);
	}
	dfs(0);
	for(int i=n+1;i<=n+m;i++){
		int x=k-dis[i];
		if(x<0)continue;
		for(int j=1;j*j<=x;j++){
			if(x%j==0 && b[par[i]][x/j]==1)ans[i]=1;
			if(x%j==0 && b[par[i]][j]==1)ans[i]=1;
		}
	}
	for(int i=n+1;i<=n+m;i++){
		if(ans[i])
			cout<<"YES"<<'\n';
		else
			cout<<"NO"<<'\n';
	}
	return 0;
}

Subtask #1
0 / 33.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2392 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 2652 KB Output is correct
2 Correct 19 ms 2796 KB Output is correct
3 Incorrect 18 ms 2652 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 34.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2392 KB Output isn't correct
2 Halted 0 ms 0 KB -