刷题记录:牛客NC13885Music Problem

传送门:牛客

题目描述:

原题是英文且十分难懂(讲了一大堆pi话,还没讲清楚),翻译成人话就是给你一堆物品,问你拿不为0的任意个物品能不能被3600整除

弄清楚题目之后我们就可以开做了,首先分析一下题目我们显然会想到一个简单的dp思维,就是开一个$dp[i][j]$记录我们的前$i$个物品中$对于3600取模为j的选择$是否存在,那么显然的最后我们的答案只要判断$dp[n][0]$是否不为0即可

那么我们该怎么转移呢,对于遇到的每一个物品$i$,显然我们的$dp[i][a[i]$%3600$]=1(假设我们之前的都不选取)$,然后我们可以枚举前一个的每一个模数(从0开始到3600),如果其模数存在的话,显然我们有$dp[i-1][j]=1(不选取第i个数字)$,并且我们应该有$dp[i][(j+a[i])$%3600$]=1(假设我们选取第i个数字的话)$

自此我们的转移过程就结束了,但是我们还需要对我们的代码进行一点优化,因为我们的$n$是达到了$1e5$的,直接枚举肯定会超时,我们会发现其实当我们的$n$的值大于$3600$的时候肯定是可以的,因为此时我们直接选取$3600$个物品就可以了,此时可以保证我们的物品的总和一定是$3600$的倍数

下面是具体的代码部分:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
int n;
int dp[3608][3608];
int a[maxn];
const int mod=3600;
int main() {
	int T;T=read();
	while(T--) {
		memset(dp,0,sizeof(dp));
		memset(a,0,sizeof(a));
		n=read();
		for(int i=1;i<=n;i++) a[i]=read();
		if(n>3600) {
			printf("YES\n");
			continue;
		}
		for(int i=1;i<=n;i++) {
			dp[i][a[i]%mod]=1;
			for(int j=0;j<mod;j++) {
				if(dp[i-1][j]) {
					dp[i][j]=1;
					dp[i][(j+a[i])%mod]=1;
				}
			}
		}
		if(dp[n][0]) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}