题目
题目链接:https://codeforces.com/problemset/problem/949/E
给出 (n) 个需要表示的数,你需要用最少的 (2^k) 或 (-2^k),使得能拼出所有需要表示的数。输出方案。
(n,|a_i|leq 10^5)。
思路
首先 (2^k) 与 (-2^k) 中最多选一个。因为选两个 (2^k) 不如选 (2^k) 和 (2^{k+1});选 (2^k) 和 (-2^k) 不如选 (2^{k+1}) 和 (-2^k)。
从低位往高位依次考虑。如果这一位中存在二进制下为 (1) 的数,那么必然会选,否则必然不选。
直接枚举所有情况跑是 (O(nA)) 的。其中 (A) 是值域。观察到处理完第 (k) 位后,剩余的所有数 (mod 2^k) 都应该为 (0)。也就是每选一个数,剩余的不同的数的数量就会减半。
那么每次操作完后都去重一次就好了。
时间复杂度 (O(nlog A))。
代码
#include <bits/stdc++.h>
using namespace std;
const int N=100010,LG=18;
int n,m,a[LG+1][N],b[LG+1],c[LG+1];
void solve(int dep,int n,int cnt)
{
if (cnt>=m || dep>LG) return;
if (n==1 && !a[dep][1])
{
memcpy(b,c,sizeof(b)); m=cnt;
return;
}
bool flag=0;
for (int i=1;i<=n;i++)
if (a[dep][i]&1) { flag=1; break; }
if (!flag)
{
for (int i=1;i<=n;i++)
a[dep+1][i]=(a[dep][i]>>1);
solve(dep+1,n,cnt);
}
else
{
for (int i=1;i<=n;i++)
a[dep+1][i]=((a[dep][i]-(a[dep][i]&1))>>1);
int n1=unique(a[dep+1]+1,a[dep+1]+1+n)-a[dep+1]-1;
c[cnt+1]=(1<<dep);
solve(dep+1,n1,cnt+1);
for (int i=1;i<=n;i++)
a[dep+1][i]=((a[dep][i]+(a[dep][i]&1))>>1);
n1=unique(a[dep+1]+1,a[dep+1]+1+n)-a[dep+1]-1;
c[cnt+1]=-(1<<dep);
solve(dep+1,n1,cnt+1);
}
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&a[0][i]);
sort(a[0]+1,a[0]+1+n);
n=unique(a[0]+1,a[0]+1+n)-a[0]-1;
m=1e9;
solve(0,n,0);
cout<<m<<"
";
for (int i=1;i<=m;i++)
cout<<b[i]<<" ";
return 0;
}