ACM_51nod 1354 选数字

1354 选数字

基准时间限制：1 秒 空间限制：131072 KB 分值: 80 难度：5级算法题

问题描述

当给定一个序列a[0],a[1],a[2],...,a[n-1] 和一个整数K时，我们想找出,有多少子序列满足这么一个条件：把当前子序列里面的所有元素乘起来恰好等于K。

样例解释

对于第一个数据，我们可以选择[3]或者[1(第一个1), 3]或者[1(第二个1), 3]或者[1,1,3]。所以答案是4。

Input

多组测试数据。在输入文件的第一行有一个整数T(0< T <= 20)，表示有T组数据。
接下来的2*T行，会给出每一组数据
每一组数据占两行，第一行包含两个整数n, K(1<=n<=1000,2<=K<=100000000)他们的含意已经在上面提到。
第二行包含a[0],a[1],a[2],...,a[n-1] (1<= a[i]<=K) 以一个空格分开。
所有输入均为整数。

Output

对于每一个数据，将答案对1000000007取余之后输出即可。

Input示例

2
3 3
1 1 3
3 6
2 3 6

Output示例

4
2

题目分析

使用map数据结构记录乘积为first value的方案数second value，最后结果为map[k];
注意map中first的值只可能为k的因数，可依此简化

AC代码

#include <iostream>
using namespace std;
#define ll long long int
#include<map>
const ll mod = 1e9+7;
map<int, int> mapint;
map<int, int> temmap;
map<int, int>::iterator it;
int main()
{
	int t;
	int n;
	ll k;
	ll num[1010];
	ll ans = 0;
	cin >> t;
	while (t--) {
		mapint.clear();
		cin >> n >> k;
		for (int i = 0; i < n; ++i) {
			cin >> num[i];
		}
		for (int i = 0; i < n; ++i) {
			if(k%num[i]==0){
				temmap = mapint;
				for (it = temmap.begin(); it != temmap.end(); ++it) {
					ll cnt = (it->first)*num[i];
					if (k %cnt == 0) {
						mapint[cnt] = (it->second + mapint[cnt]) % mod;
					}
				}
				mapint[num[i]] = (mapint[num[i]] + 1) % mod;
			}
		}
		cout << mapint[k] << endl;
	}
	return 0;
}