Codeforces Round #851 (Div. 2)

比赛链接

A. One and Two（思维题）

题意：

给定 n 个数,每个数可能为1可能为2，找到最小的k满足：前k项的乘积等于后面的项的乘积.

思路：

如果有奇数个2的话，无法找到满足条件的 k ，如果有偶数个2，那么找到一半数量的2，结束循环，输出下标即可，对于全是1的序列也满足第二个分类. 

参考代码：

#include <bits/stdc++.h>
using ll = long long;

void solve() {
    int n;
    std::cin >> n;
    int cnt = 0, ans = 0;
    std::vector<int> a(n);
    for (int i = 0; i < n; i++) {
        std::cin >> a[i];
        cnt += a[i] == 2;
    }
    if (cnt & 1) {
        std::cout << "-1\n";
        return;
    }
    for (int i = 0; i < n; i++) {
        if (a[i] == 2)
            ans++;
        if (ans == (cnt >> 1)) {
            std::cout << i + 1 << '\n';
            return;
        }
    }
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int t;
    std::cin >> t;
    while (t--) {
        solve();
    }
    return 0;
}