洛谷P1356 数列的整除性 (数学 + DP)

2020.6.2
今天新电脑到货的第一天,昨天订的今天就到了,真的够快。
用起来还可以,这次没装360,感觉也没什么问题。一拿到电脑首先下载了clion和mingw,可能因为我爱的深沉,以前都是首先下载cf来着。。。。

然后过了一道题,这道题说真的我没太想出来怎么搞,可能两天没动就迟钝了不少。题意是放置n个加号或者减号能不能获得被k整除的和。常规方法肯定是枚举,放+或者-,然而显而易见这样的复杂度大概是2^n的,超过20就悬了,更何况这是1e4的数据,看到k有点小,这道题的突破口肯定在k。那么整除的定义是%除数 == 0, 所以转移的变量除了确定的下标i之外我们又能确定了,余数j。对于每一个f[i][j] 我们都有确定的关系: i可以从i - 1转移过来, j 可以从j - a[i] 和 j + a[i] 对k取模的转移过来(k还是小) ,之后判断下能不能转移到n 和 0,代表前n个数可以凑齐对k取模为0的和,这样目的就达到了。

代码:

#include <bits/stdc++.h>
using namespace std;
#define limit (10000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%lld\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}
int n , k;
int f[limit][limit/10],a[limit];//加减各一个
int mod(int num){
    return (num + k) % k;
}
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    int kase = read();
    while(kase--){
        memset(f, 0 , sizeof(f));
        n = read(), k = read();//对着k取模
        rep(i ,1, n){
            a[i] = read();
        }
        f[1][mod(a[1])] = f[1][mod(-a[1])] = 1;//这个可以
        rep(i , 2, n){
            rep(j ,0 , k){
                f[i][j] = f[i-1][mod(j - a[i])] | f[i - 1][mod(j + a[i])];
            }
        }
        puts(f[n][0] ? "Divisible" : "Not divisible");
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章