前提・実現したいこと
AtcoderのABC50のC問題について,作成したプログラムが通らない理由がわかりません.
使用言語はC++です.
【以下問題文】
1~N までの番号がついた、N 人の人がいます。 彼らは昨日、ある順番で左右一列に並んでいましたが、今日になってその並び方が分からなくなってしまいました。 しかし、彼らは全員、「自分の左に並んでいた人数と自分の右に並んでいた人数の差の絶対値」を覚えています。 彼らの報告によると、人 i の、「自分の左に並んでいた人数と自分の右に並んでいた人数の差の絶対値」は Ai です。
彼らの報告を元に、元の並び方が何通りあり得るかを求めてください。 ただし、答えは非常に大きくなることがあるので、109+7 で割った余りを出力してください。 また、彼らの報告が間違っており、ありうる並び方がないこともありえます。 その際は 0 を出力してください。
制約
1≦N≦10^5
0≦Ai≦N−1
入力
入力は以下の形式で標準入力から与えられる。
N
A1 A2 … AN
出力
元の並び順としてありうるものが何通りあるか求め、10^9+7 で割った余りを出力せよ。
【方針】
「自分の左に並んでいた人数と自分の右に並んでいた人数の差の絶対値」(以下Aとする)をオフセットとして,Aの値がそれぞれ何回ずつ現れるかを配列で管理する.
Nが偶数か奇数かで場合わけをする.
条件を満たす場合は2^(int)(n/2)通り,条件を満たさない場合は0を出力する.
発生している問題・エラーメッセージ
テストセットの一部でWAになる.
該当のソースコード
#include<iostream>
#include<string>
#include <cmath>
using namespace std;
typedef long long ll;
int main(){
ll ans = 1;
int a[100100];
const int MOD = 1e9+7;
int n;
cin >> n;
for(int i = 0; i < n; i++) {int p; cin >> p; a[p]++;}
if(n % 2 == 1) {
if(a[0] != 1) ans = 0;
else for(int i = 2; i < n; i+=2)
if(a[i] == 2) (ans *= 2) %= MOD;
else {cout << 0 << endl; return 0;}
}
else{
for(int i = 1; i < n; i += 2)
if(a[i] == 2) (ans *= 2) %= MOD;
else {cout << 0 << endl; return 0;}
}
cout << ans << endl;
return 0;
}
回答1件
あなたの回答
tips
プレビュー