test/yukicoder/4569.test.cpp

• View this file on GitHub
• Last update: 2021-03-25 09:21:12+09:00
• Problem: https://yukicoder.me/problems/4569

Depends on

• io/single.cpp
• mod/rint.cpp

Code

// verification-helper: PROBLEM https://yukicoder.me/problems/4569

#include<bits/stdc++.h>
using namespace std;

#define call_from_test
#include "../../io/single.cpp"
#include "../../mod/rint.cpp"
#undef call_from_test

signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  int p,n;
  cin>>p>>n;
  auto as=read(n);
  string s;
  cin>>s;

  using R = Rint<int>;
  R::set_mod(p);
  R ans(as[0]);
  for(int i=1;i<n;i++){
    if(s[i-1]=='+') ans+=R(as[i]);
    if(s[i-1]=='-') ans-=R(as[i]);
    if(s[i-1]=='*') ans*=R(as[i]);
    if(s[i-1]=='/') ans/=R(as[i]);
  }

  cout<<ans<<endl;
  return 0;
}

#line 1 "test/yukicoder/4569.test.cpp"
// verification-helper: PROBLEM https://yukicoder.me/problems/4569

#include<bits/stdc++.h>
using namespace std;

#define call_from_test
#line 1 "io/single.cpp"

#line 3 "io/single.cpp"
using namespace std;
#endif

//BEGIN CUT HERE
template<typename T=int>
vector<T> read(size_t n){
  vector<T> ts(n);
  for(size_t i=0;i<n;i++) cin>>ts[i];
  return ts;
}
//END CUT HERE
#ifndef call_from_test
signed main(){
  return 0;
}
#endif
#line 1 "mod/rint.cpp"

#line 3 "mod/rint.cpp"
using namespace std;
#endif
//BEGIN CUT HERE
template<typename T>
struct Rint{
  inline static T mod;
  static void set_mod(T nmod){mod=nmod;}

  T v;
  Rint():v(0){}
  Rint(signed v):v(v){}
  Rint(long long t){v=t%mod;if(v<0) v+=mod;}

  Rint pow(long long k){
    Rint res(1),tmp(v);
    while(k){
      if(k&1) res*=tmp;
      tmp*=tmp;
      k>>=1;
    }
    return res;
  }

  static Rint add_identity(){return Rint(0);}
  static Rint mul_identity(){return Rint(1);}

  Rint inv(){return pow(mod-2);}

  Rint& operator+=(Rint a){v+=a.v;if(v>=mod)v-=mod;return *this;}
  Rint& operator-=(Rint a){v+=mod-a.v;if(v>=mod)v-=mod;return *this;}
  Rint& operator*=(Rint a){v=1LL*v*a.v%mod;return *this;}
  Rint& operator/=(Rint a){return (*this)*=a.inv();}

  Rint operator+(Rint a) const{return Rint(v)+=a;}
  Rint operator-(Rint a) const{return Rint(v)-=a;}
  Rint operator*(Rint a) const{return Rint(v)*=a;}
  Rint operator/(Rint a) const{return Rint(v)/=a;}

  Rint operator-() const{return v?Rint(mod-v):Rint(v);}

  bool operator==(const Rint a)const{return v==a.v;}
  bool operator!=(const Rint a)const{return v!=a.v;}
  bool operator <(const Rint a)const{return v <a.v;}
};
template<typename T>
ostream& operator<<(ostream &os,Rint<T> m){os<<m.v;return os;}
//END CUT HERE
#ifndef call_from_test
signed main(){
  return 0;
}
#endif
#line 9 "test/yukicoder/4569.test.cpp"
#undef call_from_test

signed main(){
  cin.tie(0);
  ios::sync_with_stdio(0);

  int p,n;
  cin>>p>>n;
  auto as=read(n);
  string s;
  cin>>s;

  using R = Rint<int>;
  R::set_mod(p);
  R ans(as[0]);
  for(int i=1;i<n;i++){
    if(s[i-1]=='+') ans+=R(as[i]);
    if(s[i-1]=='-') ans-=R(as[i]);
    if(s[i-1]=='*') ans*=R(as[i]);
    if(s[i-1]=='/') ans/=R(as[i]);
  }

  cout<<ans<<endl;
  return 0;
}

Back to top page