AtCoder Beginner Contest73

Question

#AtCoder Beginner Contest73

質問の問題

最初、何も書いていない紙がある。

joisinoお姉ちゃんが一つの数字を言うので、その数字が紙に書いてあれば紙からその数字を消し、書いていなければその数字を紙に書く。これを N回繰り返す。その後、紙に書かれている数字がいくつあるかを答える。joisinoお姉ちゃんが言った数字が、言った順番に A1,...,ANとして与えられるので、ゲーム終了後に紙に書かれている数字がいくつあるか答えてください。

この問題について取り組んでいましたがACを出すことが出来ません。
私の書いた実装のどの部分に問題があるのかをご教示いただけると幸いです。

考えとしてはsortした配列をforループで回して、
v[i] == v[i+1]つまり偶数個であればcountをインクリメントせずに i++として
2つ先のindexに移るという意図でこのコードを書きました。

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const long long INF = 1LL << 60;

int main() {
    int N, count = 0;
    cin >> N;
    vector<int> v(N);

    for(int i=0; i<N; i++) {
        cin >> v[i];
    }
    sort(v.begin(), v.end());

    for(int i=0; i<N-1; i++) {
        if(v[i] == v[i+1]) {
            i++;
        }
        else {
            count++;
        }
    }
    cout << count << endl;
}

上の実装で
入力値として以下の入力をした時は正常な出力がされていることは確認しました。

3 6 2 6 -> 1が出力
4 2 5 5 2 -> 0が出力

しかしながらWAの入力もあるためこの実装の問題点を指摘いただけると幸いです。
よろしくお願いいたします。

Answer 1

入力が1個だけのとき、0となります。

[別解] 問題のとおり、愚直/素直に実装

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    std::vector<int> v;

    int N;
    std::cin >> N;
    for(int i=0; i<N; i++) {
        int data;
        std::cin >> data;
        auto pos = std::find(v.begin(),v.end(),data);
        if ( pos == v.end() ) {
            // なかったら書き加える
            v.push_back(data);
        } else {
            // あったら消去する
            v.erase(pos);
        }
    }
    std::cout << v.size() << std::endl;
}

Comments

[encho]

ループを以下の通りにすることで解決できました。ありがとうございます。

for(int i=0; i<N; i++) {
if(v[i] == v[i+1] && i <= N-2) {
i++;
}
else {
count++;
}
}