Submission #566909

#	Time	Username	Problem	Language	Result	Execution time	Memory
566909		jamezzz	Jail (JOI22_jail)	C++17	72 / 100	5100 ms	923844 KiB

jail

#include <bits/stdc++.h>
using namespace std;

#ifdef DEBUG
#define dbg(...) printf(__VA_ARGS__);
#define getchar_unlocked getchar
#else
#define dbg(...)
#endif
#define sf scanf
#define pf printf
#define fi first
#define se second
#define pb push_back
#define li 2*i+1
#define ri 2*i+2
#define sz(x) (int)x.size()
#define mnto(x,y) x=min(x,(__typeof__(x))y)
#define mxto(x,y) x=max(x,(__typeof__(x))y)
#define INF 1023456789
#define LINF 1023456789123456789
#define all(x) x.begin(), x.end()
#define disc(x) sort(all(x));x.resize(unique(all(x))-x.begin());
typedef long long ll;
typedef long double ld;
typedef pair<int, int> ii;
typedef pair<ll, ll> pll;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<pll> vll;
mt19937 rng(time(0));

#define mod 1000000007

inline int rd(){
	int x=0;
	char ch=getchar_unlocked();
	while(!(ch&16))ch=getchar();//keep reading while current character is whitespace
    while(ch&16){//this will break when ‘\n’ or ‘ ‘ is encountered
		x=(x<<3)+(x<<1)+(ch&15);
		ch=getchar_unlocked();
	}
	return x;
}

#define maxn 120005

int n,m,s[maxn],t[maxn],sid[maxn],tid[maxn],deg[maxn];
int par[maxn],dep[maxn];
vi AL[maxn];
vector<iii> upds[maxn];
vi AL2[maxn],AL3[maxn];

void dfs(int u){
	for(int v:AL[u]){
		if(v==par[u])continue;
		par[v]=u;
		dep[v]=dep[u]+1;
		dfs(v);
	}
}

int main(){
	int tc;sf("%d",&tc);
	while(tc--){
		sf("%d",&n);
		for(int i=1;i<=n;++i){
			sid[i]=tid[i]=-1;
			par[i]=0;
			AL[i].clear();
		}
		for(int i=0;i<m;++i){
			AL2[i].clear();
			AL3[i].clear();
		}
		
		for(int i=0;i<n-1;++i){
			int a,b;sf("%d%d",&a,&b);
			AL[a].pb(b);AL[b].pb(a);
		}
		sf("%d",&m);
		for(int i=0;i<m;++i){
			sf("%d%d",&s[i],&t[i]);
			sid[s[i]]=i;
			tid[t[i]]=i;
		}
		dfs(1);
		
		//AL2[u] -> v means v has to come before u
		for(int i=0;i<m;++i){
			int u=s[i],v=t[i];
			while(u!=v){
				if(dep[u]<dep[v])swap(u,v);
				if(sid[u]!=-1&&sid[u]!=i){
					AL2[i].pb(sid[u]);
					AL3[sid[u]].pb(i);
				}
				if(tid[u]!=-1&&tid[u]!=i){
					AL2[tid[u]].pb(i);
					AL3[i].pb(tid[u]);
				}
				u=par[u];
			}
			if(sid[v]!=-1&&sid[v]!=i){
				AL2[i].pb(sid[v]);
				AL3[sid[v]].pb(i);
			}
			if(tid[v]!=-1&&tid[v]!=i){
				AL2[tid[v]].pb(i);
				AL3[i].pb(tid[v]);
			}
		}
		
		vector<int> v;
		for(int i=0;i<m;++i){
			disc(AL2[i]);
			disc(AL3[i]);
			deg[i]=AL2[i].size();
			if(deg[i]==0)v.pb(i);
		}
		
		int proc=0;
		while(!v.empty()){
			int x=v.back();
			v.pop_back();
			++proc;
			for(int y:AL3[x]){
				--deg[y];
				if(deg[y]==0)v.pb(y);
			}
		}
		
		if(proc!=m)pf("No\n");
		else pf("Yes\n");
		
	}
}

Compilation message (stderr)

jail.cpp: In function 'int main()':
jail.cpp:66:11: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   66 |  int tc;sf("%d",&tc);
      |           ^
jail.cpp:68:5: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   68 |   sf("%d",&n);
      |     ^
jail.cpp:80:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   80 |    int a,b;sf("%d%d",&a,&b);
      |              ^
jail.cpp:83:5: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   83 |   sf("%d",&m);
      |     ^
jail.cpp:85:6: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   85 |    sf("%d%d",&s[i],&t[i]);
      |      ^

• Subtask 1: 0 / 5
• Subtask 2: 5 / 5
• Subtask 3: 16 / 16
• Subtask 4: 28 / 28
• Subtask 5: 12 / 12
• Subtask 6: 11 / 11
• Subtask 7: 0 / 23