[Codeforces 835E. The penguin's game]构造+二进制

[Codeforces 835E. The penguin’s game]构造+二进制

分类：constructive algorithms

1. 题目链接

[Codeforces 835E. The penguin’s game]

2. 题意描述

一道交互题。有$n$个数，其中$n - 2$个的值为$x$， $2$个值为$y$。你可以查询至多19次。每次询问选出$c$个数,下标分别为$p_1,p_2,\dots,p_c$，程序会告诉你这$c$个数的异或和。
最后，你输出值为$y$的$2$个数的下标。
数据范围：$(2 ≤ n ≤ 1000, 1 ≤ x, y ≤ 109, x ≠ y)$。

3. 解题思路

我最开始的思路（错误的思路，可跳过）是：首先将$n$个数随机分成两个集合$s1, s2$，那么有如下3种情况：

1. 两个值为$y$的数都在$s1$中；
2. 两个值为$y$的数都在$s2$中；
3. 一个值为$y$的数在$s1$中，另外一个在$s2$中。

对于Solution3， 就可以对集合$s1, s2$分别进行一次二分$(1\le |s1|, |s2|\le \frac{n}{2})$，那么总共最多$2*log_2\lceil\frac{n}{2}\rceil$次询问就可以求出两个位置。
而Solution 3的出现概率为$\frac{1}{2}$。那么枚举$t$次，出现一次Solution 3的概率为$\frac{2^t-1}{2^t}$。
所有总共需要询问的次数为$2*log_2\lceil\frac{n}{2}\rceil+t\approx18+t$。
然后题目查询最多19次。。。GG。无限WA。

官方题解给的做法很巧妙。$10$个二进制位可以表示$[1,1000]$内所有数。假设两个$y$的下标分别为$p1, p2$。
枚举每一位，假设当前枚举第$i$位，将$[1,n]$中，二进制展开后第$i$位为$1$的数分成一个集合$s1$，第$i$位为$0$的分为一个集合$s2$，然后，查询集合$s1$的异或和，会有两种情况：

1. 如果异或和为$y,或者x\bigoplus y$,那么说明集合$s1,s2$中分别含有一个$y$，也就是说$p1,p2$在第$i$位不同，保存当前位$i$；
2. 如果异或和为$0,或者x$,两个值为$y$的数都在$s1$中或者都在$s2$中，也就是说$p1,p2$在第$i$位相同。

在第一个情况中，可以根据第$i$位将两个$y$划分到两个集合中去。任意一种划分，对集合$s1$、或者$s2$进行二分，就可以求出一个$y$的下标。【这里要注意的是，选择集合$s1、s2$中较小的那个集合进行二分】
假设现在一直了$p1$，又因为之前在枚举过程中，求出了所有$p1,p2$二进制在哪一位不同。所以在$p1$的对应位取反就得到了$p2$。
这样可以保证在$\lceil log_2n\rceil+log_2\lceil\frac{n}{2}\rceil \le10+9=19$次求出答案。

Orz…神奇的做法！神奇的二进制！

补充：其实，这种二进制的思路在其他地方也用得很普遍。e.g: [Matrix67趣题：老鼠与毒药问题的推广]

4. 实现代码

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

typedef long long LL;
typedef long double LB;
typedef unsigned int uint;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
typedef pair<LB, LB> PLB;
typedef vector<int> VI;

const int INF = 0x3f3f3f3f;
const LL INFL = 0x3f3f3f3f3f3f3f3fLL;
const long double PI = acos(-1.0);
const long double eps = 1e-4;
void debug() { cout << endl; }
template<typename T, typename ...R> void debug (T f, R ...r) { cout << "[" << f << "]"; debug (r...); }
template<typename T> inline void umax(T &a, T b) { a = max(a, b); }
template<typename T> inline void umin(T &a, T b) { a = min(a, b); }
template <typename T> inline bool scan_d (T &ret) {
    char c; int sgn;
    if (c = getchar(), c == EOF) return 0; //EOF
    while (c != '-' && (c < '0' || c > '9') ) if((c = getchar()) == EOF) return 0;
    sgn = (c == '-') ? -1 : 1;
    ret = (c == '-') ? 0 : (c - '0');
    while (c = getchar(), c >= '0' && c <= '9') ret = ret * 10 + (c - '0');
    ret *= sgn;
    return 1;
}
template<typename T> void print(T x) {
    static char s[33], *s1; s1 = s;
    if (!x) *s1++ = '0';
    if (x < 0) putchar('-'), x = -x;
    while(x) *s1++ = (x % 10 + '0'), x /= 10;
    while(s1-- != s) putchar(*s1);
}
template<typename T> void println(T x) { print(x); putchar('\n'); }
template<typename T> T randIntv(T a, T b) { return rand() % (b - a + 1) + a; } /*[a, b]*/

const int MAXN = 1005;
uint seed = 131;

LL x, y, q;
int times;
void get_q(int a[], int len);

void query(int a[], int len) {
    putchar('?'); putchar(' '); print(len);
    for(int i = 1; i <= len; ++i) { putchar(' '); print(a[i]); } putchar('\n'); fflush(stdout);
    scan_d(q); -- times;
//    get_q(a, len); -- times;      /****/
    if(q == -1) exit(0);
}

int work(int a[], int len) {
    int lb = 1, ub = len, md, p;
    while(lb <= ub) {
        md = (lb + ub) >> 1;
        query(a, md);
        if(q != 0 && q != x) p = md, ub = md - 1;
        else lb = md + 1;
    }
    return a[p];
}
int p1, p2;

int n, r[15][MAXN];
bool in_s1[MAXN];


LL Ans[MAXN];

void getAns()  {
    Ans[1] = Ans[2] = y;
    for(int i = 3; i <= n; ++i) Ans[i] = x;
    shuffle(Ans + 1, Ans + n, default_random_engine(seed));
}
void get_q(int a[], int len) {
    q = 0;
    for_each(a + 1, a + len + 1, [](int item) { q ^= Ans[item]; });
}

int main() {
#ifdef ___LOCAL_WONZY___
    freopen ("input.txt", "r", stdin);
#endif // ___LOCAL_WONZY___
    scan_d(n), scan_d(x), scan_d(y);
    n = randIntv(700, 1000); x = rand() + 1;            /****/
    do {y = rand() + 1;} while(x == y); getAns();       /****/
    times = 19; p1 = p2 = -1;
    int s1 = -1, s2, temp = 0;
    for(int j = 0; j < 10; ++j) {
        r[j][0] = 0;
        for(int i = 1; i <= n; ++i) {
            if(i >> j & 1) r[j][++ r[j][0]] = i;
        }
        if(r[j][0] == 0) continue;
        query(r[j], r[j][0]);
        if(q != 0 && q != x) {
            temp |= 1 << j;
            s1 = j;
        }
    }

    r[s2 = 10][0] = 0;
    for(int i = 1; i <= n; ++i) in_s1[i] = false;
    for(int i = 1; i <= r[s1][0]; ++i) in_s1[r[s1][i]] = true;
    for(int i = 1; i <= n; ++i) if(!in_s1[i]) r[s2][++ r[s2][0]] = i;
    if(r[s1][0] > r[s2][0]) swap(s1, s2);

    p1 = work(r[s1], r[s1][0]);

    p2 = p1 ^ temp;
    if(p1 > p2) swap(p1, p2);

    putchar('!'); putchar(' '); print(p1); putchar(' '); print(p2); putchar('\n'); fflush(stdout);
//        debug("times: ", times, n, x, y, p1, p2, Ans[p1], Ans[p2]);
//        while(times < 0);
//    assert(times >= 0);
//    assert(Ans[p1] == Ans[p2] && Ans[p1] == y);
//    }
#ifdef ___LOCAL_WONZY___
    cout << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC * 1000 << " ms." << endl;
#endif // ___LOCAL_WONZY___
    return 0;
}