atcoder エイジングプログラミングコンテスト D問題が解けない。

Question

atcoder エイジングプログラミングコンテスト D問題が解けず、解説を見て実装してみたのですが、うまくいきません。

https://atcoder.jp/contests/aising2020/tasks/aising2020_d

いろいろ試行錯誤し、RE や TLE は回避できたのですが、WA が出てしまいます。

よければ下のコードで何がおかしいのか指摘してもらえるとうれしいです。よろしくお願いします。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string>
#include<map>
#include<vector>
#define rep(i, n) for(int i=0; i<n; i++)
#define rep_one(i, n) for(int i=1; i<n; i++)
using namespace std;
typedef long long ll;

int solve(int x, int count){
    if(x == -1) return 0;
    if(x == 0) return count;
    int y = x;
    int pc = 0;
    while(y != 0){
        pc += y%2;
        y /= 2;
    }
    return solve(x%pc, count+1);
}

int main(){
    int n; cin >> n;
    string x; cin >> x;
    ll xp=0, xm=0;
    int pc=0;
    int x_list[n];
    rep(i, n){
        x_list[i] = (int)(x[i]-'0');
        pc += x_list[i];
    }
    rep(i, n){
        xp = (xp*2)%(pc+1);
        xp += x_list[i];
        if(pc-1 > 0){
            xm = (xm*2)%(pc-1);
            xm += x_list[i];
        }
    }
    int xi[n];
    ll tmp=1;
    rep(i, n) tmp *= 2;
    rep(i, n){
        tmp /= 2;
        if(x_list[i]){
            if(pc-1 < 1){xi[i]=-1; continue;}
            xi[i] = (xm-tmp+pc-1)%(pc-1);
        }else{
            xi[i] = (xp+tmp)%(pc+1);
        }
    }
    rep(i, n){
        cout << solve(xi[i], 1) << endl;
    }
}

Answer 1

    ll tmp=1;
    rep(i, n) tmp *= 2;

ここがオーバーフローします

Comments

[goro_gnm]

変な勘違いをし、オーバーフローはないと思っていました、ありがとうございます！