“蓝桥杯”练习系统 ALGO-194 试题 算法训练 审美课

试题 算法训练 审美课

资源限制

时间限制：1.0s 内存限制：256.0MB

问题描述

《审美的历程》课上有n位学生，帅老师展示了m幅画，其中有些是梵高的作品，另外的都出自五岁小朋友之手。老师请同学们分辨哪些画的作者是梵高，但是老师自己并没有答案，因为这些画看上去都像是小朋友画的……老师只想知道，有多少对同学给出的答案完全相反，这样他就可以用这个数据去揭穿披着皇帝新衣的抽象艺术了（支持帅老师^_）。
　　答案完全相反是指对每一幅画的判断都相反。

输入格式

第一行两个数n和m，表示学生数和图画数； 　　
接下来是一个n*m的01矩阵A：
如果aij=0，表示学生i觉得第j幅画是小朋友画的； 　　
如果aij=1，表示学生i觉得第j幅画是梵高画的。

输出格式

输出一个数ans：表示有多少对同学的答案完全相反。

样例输入

3 2
1 0
0 1
1 0

样例输出

2

样例说明

同学1和同学2的答案完全相反；
同学2和同学3的答案完全相反；
所以答案是2。

数据规模和约定

对于50%的数据：n<=1000；
对于80%的数据：n<=10000；
对于100%的数据：n<=50000，m<=20。

---

说实话是抱着试试看的态度，结果意外顺利地通过了。果然数组可以豪爽的开很大很大。

思路详解

观察给定的数据范围，n 的数字较大，对于我们保存数据的查找、插入效率有相当的要求。我们每一次查找与一个同学的答案完全相反的答案的效率是关键。时间限制 1s 的情况下，我们优先考虑能在 O(1)时间完成查找任务的数据存储方式。

我们发现 m 最大不超过 20 ，结合每一个同学对于一幅画只有两种态度，以及我们需要统计完全相反的答案，我们可以使用 「状态压缩」的技巧，把每一个同学的答案用一个整数型变量表示，最多需要用到其中的 20 个位，每一位上的 0 或 1 对应同学对于一幅画的态度。那么这个整数可以唯一对应一个可能由同学给出的答案。

这样做的好处是，整数值可以直接作为数组的下标。且在 O(1) 时间内，我们可以直接计算出相反答案对应的下标值。这一步操作可以藉由 异或 来完成，即每一位上与 1 做异或，达到反转效果；或者用一个每一位满载着 1 的整数减去现有的下标，这样一来，只有作为减数的下标为 0 的位上会留下 1。我们为了方便理解，采用了第二种写法。

为了构造这样一个 m 位的 「满载整数」，可以把 1 左移 m 位，得到右起 m 位都是 0， 右起 m + 1位是 1 的整数,。将它减去 1，右起 m 位上变成 1 ，就得到了需要的整数。

由于答案要求我们求得意见相反同学的对数，我们根据组合的原理，对于一对相反的意见，我们需要统计的是两种意见同学数量相乘。我们每一次遍历到一种新的意见，就把它和与它对立的意见分别的同学数相乘加到答案。这之后，把相反的意见的同学数重置为 0 ，避免二次统计。

---

代码

#include <iostream>
using namespace std;
const int M = 1<<22;    // max m
int freq[M];            // frequency
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n, m;
    cin >> n >> m;
    for (int i=0; i<n; ++i) {
        int pos = 0;    // index of freq[]
        for (int j=0; j<m; ++j) {   // loads bit by bit
            int c;
            cin >> c;
            pos <<= 1;
            pos += c;
        }
        ++freq[pos];
    }
    int ans = 0;
    int ful = (1<<m) - 1;       // fully loaded '1'
    for (int i=0; i<(1<<m); ++i) {
        if (freq[i] && freq[ful - i]) {
            // cout << i << ',' << ful - i << endl;
            ans += freq[i] * freq[ful - i];
            freq[ful - i] = 0;  // avoid counting again
        }
    }
    cout << ans << endl;
    return 0;
}

时间125ms； 内存6.632MB；

是不是挺好玩的呀…