Codeforces 691A. Fashion in Berland

题目链接:http://codeforces.com/problemset/problem/691/A

题意:

　　有一个夹克上有 n 个扣子,能扣紧这个夹克的要求是恰好有 n - 1 个扣子是扣着的,如果这个夹克恰有一个扣子,那么这个扣子应该是扣着的,用 “1” 代表扣着,用 “0” 代表开着,问这个夹克能否扣紧.(原题意比较难翻译！！！！)

代码:

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 100000;
typedef long long LL;

int main() {
    ios_base::sync_with_stdio(0); cin.tie(0);
    int n; cin >> n;
    int a[3] = {0};
    for(int i = 0; i < n; i++) { int m; cin >> m; a[m] ++ ; }
    int OK = 0;
    if(n == 1 && a[1] == 1) OK = 1;
    if(n != 1 && a[1] == n - 1) OK = 1;
    if(OK) cout << "YES" << endl;
    else cout << "NO" << endl;
    return 0;
}