牛客算法周练——Music Problem（dp / bitset + 鸽巢定理）

题目链接

---

题目大意

给出一个序列，从中选择若干数（至少一个）能否得到$3600$的倍数。

解题思路

可能一开始想到，怎么枚举，有那么多种情况，但是实际上也就$3600$种情况，设$dp[i]$为余数为$i$的选择，每次判断$a[i]$时，我们拿它和之前的$1-3599$种已经存在的答案相加取模保存，下面就更新每种新得到的答案，最后更新一下$dp[a[i]]$，这里提前对每个$a[i]$取模处理。为什么不能提前对$dp[a[i]]$处理或者每得到一种新的答案就更新$dp[(j+a[i])\%3600]$？就给出一组数据即可证明:$1800$，如果提前更新那么$dp[1800]=1$，那么更新$1-3599$中存在每个数时就考虑到了$1800$，因此我们的更新是和前面的所有情况合并处理，因此每一步的顺序必须确定，不能提前也不能落后

实际上本题还能使用$bitset$做，创建一个$3600$位的$bitset$，第一个数为$2000$，假设第二个数为$1000$，显然相加没有超过$3600$，我们需要标志$b[2000+1000]=1$，即当前$bitset$左移$1000$位；但是如果第二个数为$3000$，$(2000+3000)\%3600=1400$，而$2000-(3600-3000)=1400$，因此此时需要标志$b[1400]$为$1$，即将$bitset$右移$(3600-3000)$位。因为我们处理每种情况都要考虑，那么就相或即可。

PS：本题还能根据数论定理优化，当$n\ge 3600$时一定是$YES$，详见下图：

dp：(1500ms)

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=1e5+10;
int a[maxn];
bool dp[3610];
vector<int> res;

int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t,n;
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=1;i<=n;i++){
            cin>>a[i];
            a[i]%=3600;
        }
        memset(dp,0,sizeof dp);
        if(n>=3600){
            cout<<"YES"<<endl;
            continue;
        }
        for(int i=1;i<=n;i++){
            if(dp[0]) break;
            res.clear();
            for(int j=1;j<3600;j++)
                if(dp[j]) res.push_back((j+a[i])%3600);
            for(int j=0;j<res.size();j++)
                dp[res[j]]=1;
            dp[a[i]]=1;
        }
        if(dp[0]) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
}

bitset：(150ms)

#include <iostream>
#include <bitset>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=1e5+10;
int a[maxn];

int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t,n;
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=1;i<=n;i++){
            cin>>a[i];
            a[i]%=3600;
        }
        if(n>=3600){
            cout<<"YES"<<endl;
            continue;
        }
        bitset<3600> b;
        for(int i=1;i<=n;i++){
            b|=(b<<a[i])|(b>>(3600-a[i]));
            b[a[i]]=1;
        }
        if(b[0]) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
}