洛谷 P2049 魔术棋子(vector)

题目传送门

解题思路:

用一个vector维护每一个点都可以乘出哪些数来,然后将(n,m)的所有数从小到大输出即可.

要用一个bool ff[j][k]来维护当前这个点(i,j)里面有没有被放过k,以此保证vector是个集合,不然会MLE(压维也不行)

AC代码:

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>

using namespace std;

int n,m,k,a[101][101],ans,now;
vector<int> s[2][101];
bool vis[101],ff[101][101];

int main() {
    scanf("%d%d%d",&n,&m,&k);
    for(int i = 1;i <= n; i++)    
        for(int j = 1;j <= m; j++)
            scanf("%d",&a[i][j]);
    s[1][1].push_back(a[1][1]);
    for(int i = 1;i <= n; i++,now ^= 1)
        for(int j = 1;j <= m; j++) {
            for(int p = 1;p <= s[now^1][j].size(); p++) {
                int u = a[i][j] * s[now^1][j][p-1] % k;
                if(!ff[j][u]) {
                    s[now][j].push_back(u);
                    ff[j][u] = 1;
                }
            }
            s[now^1][j].clear();
            for(int p = 1;p <= s[now][j-1].size(); p++) {
                int u = a[i][j] * s[now][j-1][p-1] % k;
                if(!ff[j][u]) {
                    s[now][j].push_back(u);
                    ff[j][u] = 1;
                }
            }
            memset(ff,0,sizeof(ff));
        }
    for(int i = 1;i <= s[now^1][m].size(); i++) 
        if(vis[s[now^1][m][i-1] % k] == 0)
            vis[s[now^1][m][i-1] % k] = 1,ans++;
    printf("%d
",ans);
    for(int i = 0;i <= k; i++)
        if(vis[i])
            printf("%d ",i);
    return 0;
}