牛客算法周周练2 Music Problem(背包)

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
typedef long long ll;
ll a[maxn],f[maxn],vis[maxn];
int main()
{
    
    
	int t,n;
	cin>>t;
	while(t--)
	{
    
    
		cin>>n;
		for(int i=1;i<=n;i++)
		cin>>a[i];
		
		memset(f,0,sizeof(f));
		memset(vis,0,sizeof(vis));
		for(int i=1;i<=n;i++)
		{
    
    	
			int cnt=0;//记录此次更新的个数 
			a[i]%=3600;
			if(f[0]==1)//及时跳出 
			{
    
    
				break;
			}
			for(int j=1;j<3600;j++)
			{
    
    
				if(f[j])
				{
    
    
					vis[++cnt]=(j+a[i])%3600;//记录所有能取到的值 
				}
			}
			for(int j=1;j<=cnt;j++)
			{
    
    
				f[vis[j]]=1;//更新所有能取到的值 
			}
			f[a[i]]=1;//要放记录后面 
		}
		if(f[0])
		cout<<"YES"<<endl;
		else
		cout<<"NO"<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/xiaolan7777777/article/details/105616208