题意
n个点的树
每次可以询问两个集合S,T,S->T经过x的点对有多少个
确定树的形态
n <= 500,11111次询问
题解
显然询问集合应该是一个点到一个集合,这样才能确定更多有用信息。
发现如果是询问x到剩余点经过v的点对,可以确定出v子树内和子树外的点数
这样可以找到所有叶子,然后再找叶子的fa。删去叶子
但是这样是n2的
题解上的做法利用了更好的性质:
首先强制让根是1,那么可以通过询问1到其他点经过i,来确定i的子树大小。
然后按照sz从小到大,找儿子。
设当前还没有找到fa的集合为X,当前考虑v,
询问{{1},{X},v}可以知道X中有哪些点是v的直接儿子(因为是从小到大考虑,这些点都还没有父亲,不可能是v的子树中的其他点)
那么我们二分找到最左边的一个是x的儿子的点,删去它,再继续操作。
这样询问次数是n + nlogn的
总结:
题解的巧妙在于固定了询问的单点----看成根
然后利用sz定序,树上的性质最特殊的就是fa只有一个,所以从叶子开始考虑往往可以简化问题。
最后为了确定具体的点,二分找最左边的
#include<bits/stdc++.h>
using namespace std;
#define rep(i,l,r) for(register int i = l ; i <= r ; i++)
#define repd(i,r,l) for(register int i = r ; i >= l ; i--)
#define rvc(i,S) for(register int i = 0 ; i < (int)S.size() ; i++)
#define rvcd(i,S) for(register int i = ((int)S.size()) - 1 ; i >= 0 ; i--)
#define fore(i,x)for (register int i = head[x] ; i ; i = e[i].next)
#define forup(i,l,r) for (register int i = l ; i <= r ; i += lowbit(i))
#define fordown(i,id) for (register int i = id ; i ; i -= lowbit(i))
#define pb push_back
#define prev prev_
#define stack stack_
#define mp make_pair
#define fi first
#define se second
#define lowbit(x) ((x)&(-(x)))
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int,int> pr;
const ld inf = 2e18;
const int N = 3e6 + 10;
const int maxn = 100020;
const ll mod = 1e9 + 7;
inline ll power(ll x,ll y){
y = ((y % (mod - 1)) + (mod - 1)) % (mod - 1);
// if ( y < 0 ) return power(power(x,-y),mod - 2);
ll res = 1;
while ( y ){
if ( y & 1 ) res = res * x % mod;
x = x * x % mod;
y >>= 1;
}
return res;
}
int n;
vector <int> vec,S,T;
int sz[maxn],a[maxn],fa[maxn];
inline int query(int v){
printf("%d\n",S.size());
rvc(i,S) printf("%d ",S[i]);
printf("\n%d\n",T.size());
rvc(i,T) printf("%d ",T[i]);
printf("\n%d\n",v);
fflush(stdout);
int t;
scanf("%d",&t);
return t;
}
void getsz(int x){
S.clear() , T.clear();
S.pb(1);
rep(i,2,n) T.pb(i);
sz[x] = query(x);
}
bool check(int id,int x){
S.clear();
S.pb(1);
T.clear();
rep(i,0,id) T.pb(vec[i]);
return query(x);
}
inline void getchild(int x){
if( !vec.size() ) return;
while ( 1){
int l = 0 ,r = vec.size() - 1,id = -1;
while ( l <= r ){
int mid = (l + r) >> 1;
if ( check(mid,x) ) r = mid - 1 , id = mid;
else l = mid + 1;
}
if ( id != -1 ){
fa[vec[id]] = x;
swap(vec[id],vec.back());
vec.pop_back();
}
else break;
if ( !vec.size() ) break;
}
}
bool cmp(int x,int y){ return sz[x] < sz[y]; }
int main(){
scanf("%d",&n);
rep(i,2,n) getsz(i) , a[i - 1] = i;
sort(a + 1,a + n,cmp);
rep(i,1,n - 1){
int x = a[i];
getchild(x);
vec.pb(x);
}
printf("ANSWER\n");
rep(i,2,n){
if ( !fa[i] ) fa[i] = 1;
printf("%d %d\n",i,fa[i]);
}
}