oj.uz

Submission #899745

# Submission time Handle Problem Language Result Execution time Memory
899745 2024-01-06T23:42:29 Z YassirSalama ICC (CEOI16_icc) C++17
18 / 100
 220 ms 620 KB 
#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <iomanip>
#include <cmath>
#include <limits>
#include <map>
#include <utility>
#include <cctype>
#include <string>
#include <cstring>
#include <stack>
#include <queue>
#include<assert.h>
#include <functional>
#include <iterator>
#include "icc.h"
using namespace std;
#define OVL(x,s) for(auto y:x) cout<<y<<s; cout<<"\n";
#ifdef IOI
void dbg_out() { cout << endl; }
template<typename Head, typename... Tail> void dbg_out(Head H, Tail... T) { cout << ' ' << H; dbg_out(T...); }
#define dbg(...) cout << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__);
#else
#define dbg(...) 1337;
#endif
#define endl "\n"
#define pb push_back
#define F first
#define S second
#define ll long long
#define mod 1000000007
#define all(v) v.begin(),v.end()
int ask(vector<int> a,vector<int> b){
	int n=a.size();
	int m=b.size();
	int A[n];
	int B[m];
	for(int i=0;i<n;i++) A[i]=a[i];
	for(int i=0;i<m;i++) B[i]=b[i];
	return query(n,m,A,B);
}
const int MAXN=2e3+100;
int par[MAXN];
struct DSU {
	DSU(){
		for(int i=0;i<MAXN;i++) par[i]=i;
	}
	int find(int node){
		return node==par[node]?node:par[node]=find(par[node]);
	}
	bool same(int a,int b){
		return find(a)==find(b);
	}
	void merge(int a,int b){
		a=find(a);
		b=find(b);
		if(a==b) return;
		par[a]=b;
	}
};
int dnc(vector<int> a,vector<int> b,int n){
	if(n==0) return 0;
	if(n==1){
		return b[0];
	}
	int mid=n/2;
	vector<int> c,d;
	for(int i=0;i<mid;i++) c.pb(b[i]);
	for(int i=mid;i<n;i++) d.pb(b[i]);
	if(ask(a,c)) return dnc(a,c,c.size());
	if(ask(a,d)) return dnc(a,d,d.size());
	return 0;
}
void run(int n){
	DSU dsu=DSU();
	for(int ww=0;ww<n-1;ww++){
		int x=-1;
		int y=-1;
		for(int i=1;i<=n;i++){
			vector<int> a;
			vector<int> b;
			a.push_back(i);
			for(int j=1;j<=n;j++){
				if(i==j) continue;
				if(dsu.same(i,j)) continue;
				b.pb(j);
			}
			if(b.size()==0) continue;
			// dbg(i);
			// OVL(a," ")
			// OVL(b," ")
			// dbg(ask(a,b))
			if(ask(a,b)){
				// dbg(i)
				// OVL(b," ")
				int x=dnc(a,b,b.size());
				// dbg(i,x);
				setRoad(i,x);
				dsu.merge(i,x);
				break;
			}
		}
		continue;
		// dbg(x,y)
		setRoad(x,y);
		dsu.merge(x,y);
		// break;

		// setRoad(1,1);
	}
}

#ifdef IOI
int main(){
	int n;
	cin>>n;
	vector<pair<int,int>> v;
	for(int i=0;i<n-1;i++){
		int a,b;
		cin>>a>>b;
		v.pb({a,b});
	}
	for(int i=0;i<MMS;i++) pr[i]=i;
	EDGES=v;
	mp[{EDGES[0].F,EDGES[0].S}]=1;
	mp[{EDGES[0].S,EDGES[0].F}]=1;

	merge(EDGES[0].F,EDGES[0].S);
 	run(n);
 	cout<<"YES"<<endl;
}
#endif

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 604 KB Ok! 217 queries used.
2 Correct 9 ms 604 KB Ok! 200 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 69 ms 604 KB Ok! 1737 queries used.
2 Correct 70 ms 600 KB Ok! 1789 queries used.
3 Correct 85 ms 620 KB Ok! 1796 queries used.

Subtask #3
0 / 22.0

# Verdict Execution time Memory Grader output
1 Incorrect 220 ms 604 KB Number of queries more than 4500 out of 2250
2 Halted 0 ms 0 KB -

Subtask #4
0 / 21.0

# Verdict Execution time Memory Grader output
1 Incorrect 199 ms 620 KB Number of queries more than 4000 out of 2000
2 Halted 0 ms 0 KB -

Subtask #5
0 / 29.0

# Verdict Execution time Memory Grader output
1 Incorrect 176 ms 620 KB Number of queries more than 3550 out of 1775
2 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 163 ms 616 KB Number of queries more than 3250 out of 1625
2 Halted 0 ms 0 KB -