C - Bakry and Partitioning
给定一棵树 问是否可以通过割边是其变为成2到k个连通块 且所有连通块上的点异或起来相等
我们可以这样考虑
假如割之后的连通块个数为2x个 即偶数个 若这两个连通块的异或值相等 则整棵树的异或值应该为0
那么假如割之后的联通块个数为2x+1个 即奇数个的时候该怎么办呢 这种情况整颗树的异或值肯定不为0的 且连通块个数肯定是大于三的 那么如果最后被割为五个连通块 我们可以使三个相邻连通块合并为一个连通块 异或值依然不变 所以我们只要可以在树中找出大于等于三个连通块即可 且k还要大于等于三 用一个dfs维护一下
代码
#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); i ++ )
using namespace std;
const int N = 2e5 + 10;
inline int read()
{
register int x = 0, k = 1;
char c = getchar();
while (c > '9' || c < '0')
{
if (c == '-') k = -1;
c = getchar();
}
while (c <= '9' && c >= '0')
{
x = (x << 1) + (x << 3) + (c ^ 48);
c = getchar();
}
return x * k;
}
int n, k;
int val[N], sval, cnt;
vector<int> G[N];
void init()
{
rep(i, n) G[i].clear();
sval = 0, cnt = 0;
}
int dfs(int u, int fa)
{
int res = val[u];
for (auto j : G[u])
if (j != fa) res ^= dfs(j, u);
if (res == sval)
{
res = 0;
cnt ++ ;
}
return res;
}
void solve()
{
n = read(), k = read();
rep(i, n) val[i] = read(), sval ^= val[i];
rep(i, n - 1){
int a, b;
a = read(), b = read();
G[a].push_back(b); G[b].push_back(a);
}
dfs(1, 0);
if (cnt >= 3 && k >= 3 || sval == 0) puts("YES");
else puts("NO");
init();
}
int main()
{
int t; t = read();
while (t -- ) solve();
return 0;
}
E. Bored Bakry
题意为给定一个序列 问这个序列中最长的 各值and的和大于各值xor的和 的区间长度为多少
假如某个区间一定符合题意 那么它的长度一定为偶数长度 因为奇数长度的xor和一定大于等于and和
可以这样看这个区间 当此区间中所有元素的某一位全为1(且区间长度为偶数) 且它的高位异或和为0 则此区间是满足题意的
我们可以从高位开始遍历 每次找符合题意的区间更新最大值 用b数组来维护1-i区间1的个数 c数组来维护高位前缀异或和
代码
#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); i ++ )
using namespace std;
const int N = 1e6 + 10;
inline int read()
{
register int x = 0, k = 1;
char c = getchar();
while (c < '0' || c > '9')
{
if (c == '-') k = -1;
c = getchar();
}
while (c >= '0' && c <= '9')
{
x = (x << 3) + (x << 1) + (c ^ 48);
c = getchar();
}
return x * k;
}
int n;
int a[N], b[N], c[N];
int st[N];
void solve()
{
n = read();
rep(i, n) a[i] = read();
int ans = 0;
for (int j = 20; j >= 0; j -- )
{
rep(i, n){
b[i] = b[i - 1] + ((a[i] >> j) & 1);
c[i] = c[i - 1] ^ (a[i] >> j);
}
memset(st, -1, sizeof st);
st[0] = 0;
rep(i, n){
if (st[c[i]] != -1)
{
if (b[i] - b[st[c[i]]] == i - st[c[i]])
ans = max(ans, i - st[c[i]]);
else st[c[i]] = i;
}
else st[c[i]] = i;
}
}
printf("%d\n", ans);
}
int main()
{
solve();
return 0;
}