51nod与查询 1406

题意：

1406 与查询
2.0 秒 131,072.0 KB 80 分 5级题
有n个整数。输出他之中和x相与之后结果为x的有多少个。x从0到1，000，000

输入
第一行输入一个整数n。(1<=n<=1,000,000).
第二行有n个整数a[0],a[1],a[2],…a[n-1]，以空格分开.(0<=a[i]<=1,000,000)
输出
对于每一组数据，输出1000001行，第i行对应和i相与结果是i的有多少个数字。
输入样例
3
2 3 3
输出样例
3
2
3
2
0
0
……
后面还有很多0

思路：

正确答案当然是dp。。。为什么呢？
首先，我们可以看出来i与a[j]相与的结果还是i，那么i中的1是a[j]中的1的一部分，那么我们就是利用了a[j]中的1。。

但是我走错了路啊5555

写错了转移方程，我写的代码

const int maxn = 1e6 + 5;
const double pi = acos(-1.0);

int n;
int a[maxn];
int num[maxn][20];
int dp[maxn];
int sum[50];

int main(){
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&a[i]);
		int tmp = a[i]; int cnt = 0;
		while(tmp){
			int cur = tmp % 2;
			num[i][cnt++] = cur;
			tmp = tmp / 2;
		}
	}
	for(int i = 1;i <= n;i++){
		for(int j = 0;j <= 17;j++){
			if(num[i][j]) sum[j]++;
		}
	}
	dp[0] = n;
	for(int i = 0;i <= 20;i++){
		for(int j = 1000000;j >= (1 << i);j--){
			if(dp[j - (1 << i)]){
				dp[j] = min(dp[j - (1 << i)],sum[i]);
			}
		}
	}
	for(int i = 0;i <= 1000000;i++)
		printf("%d\n",dp[i]);
	return 0;
}

我以为只要j - (1 << i) 存在，那么j就可以由前面的转移过来，应该是考虑的情况太少了

最终结果就是对了不到一半

正解应该是什么呢？？？

对于1 ～ 1000000的每一个数只要第i位上有1，那么 dp[j - (1 << i)] = dp[j - (1 << i)] + dp[j];

代码实现：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<vector>
#include<cstdlib>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e6 + 5;
const double pi = acos(-1.0);

int n;
int a[maxn];
int num[maxn][20];
int dp[maxn];
int sum[50];

int main(){
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%d",&a[i]);
		dp[a[i]]++;
	}
	sort(a + 1,a + 1 + n);
	for(int i = 0;i <= 20;i++){
		for(int j = (1 << i);j <= a[n];j++){
			if(j & (1 << i)){
				dp[j - (1 << i)] += dp[j];
			}
		}
	}
	dp[0] = n;
	for(int i = 0;i <= 1000000;i++)
		printf("%d\n",dp[i]);
	return 0;
}