結果
問題 | No.695 square1001 and Permutation 4 |
ユーザー |
|
提出日時 | 2023-06-25 14:39:16 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 846 ms / 7,000 ms |
コード長 | 3,369 bytes |
コンパイル時間 | 2,136 ms |
コンパイル使用メモリ | 199,488 KB |
最終ジャッジ日時 | 2025-02-15 02:12:58 |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 12 |
ソースコード
#include<bits/stdc++.h>using namespace std;//#pragma GCC optimize("Ofast")#define rep(i,n) for(ll i=0;i<n;i++)#define repl(i,l,r) for(ll i=(l);i<(r);i++)#define per(i,n) for(ll i=(n)-1;i>=0;i--)#define perl(i,r,l) for(ll i=r-1;i>=l;i--)#define fi first#define se second#define pb push_back#define ins insert#define pqueue(x) priority_queue<x,vector<x>,greater<x>>#define all(x) (x).begin(),(x).end()#define CST(x) cout<<fixed<<setprecision(x)#define vtpl(x,y,z) vector<tuple<x,y,z>>#define rev(x) reverse(x);using ll=long long;using vl=vector<ll>;using vvl=vector<vector<ll>>;using pl=pair<ll,ll>;using vpl=vector<pl>;using vvpl=vector<vpl>;const ll MOD=1000000007;const ll MOD9=998244353;const int inf=1e9+10;const ll INF=4e18;const ll dy[9]={0,1,-1,0,1,1,-1,-1,0};const ll dx[9]={1,0,0,-1,1,-1,1,-1,0};template<class T> inline bool chmin(T& a, T b) {if (a > b) {a = b;return true;}return false;}template<class T> inline bool chmax(T& a, T b) {if (a < b) {a = b;return true;}return false;}//中国剰余定理// 返り値: a と b の最大公約数// ax + by = gcd(a, b) を満たす (x, y) が格納されるlong long extGcd(long long a, long long b, long long &x, long long &y) {if (b == 0) { x = 1; y = 0; return a; }long long d = extGcd(b, a%b, y, x);y -= a/b * x;return d;}//-1が帰ってくる可能性に注意pair<long long, long long> ChineseRem(const vector<long long> &b, const vector<long long> &m) {long long r = 0, M = 1;for (int i = 0; i < (int)b.size(); ++i) {long long p, q;long long d = extGcd(M, m[i], p, q); // p is inv of M/d (mod. m[i]/d)if ((b[i] - r) % d != 0) return make_pair(0, -1);long long tmp = (b[i] - r) / d * p % (m[i]/d);r += M * tmp;M *= m[i]/d;}return make_pair((r+M+M)%M, M);}int mod=592951213;int dp[10000010];int main(){ll n,m;cin >> n >> m;vl x(m);rep(i,m)cin >> x[i];vl cand;vl mods;{memset(dp,0,sizeof(dp));dp[1]=1;for(ll i=1;i<=10000005;i++){rep(j,m){if(i+x[j]<=10000005){dp[i+x[j]]+=dp[i];dp[i+x[j]]%=mod;}}}ll f=n/2;ll ans=0;rep(j,m){for(ll i=f;i>=1;i--){if(i+x[j]>f&&i+x[j]<=n){ans+=(ll)dp[i]*dp[n-i-x[j]+1];ans%=mod;}}}//cout << ans << endl;cand.emplace_back(ans);mods.emplace_back(mod);}mod=17*9920467;{memset(dp,0,sizeof(dp));dp[1]=1;for(ll i=1;i<=10000005;i++){rep(j,m){if(i+x[j]<=10000005){dp[i+x[j]]+=dp[i];dp[i+x[j]]%=mod;}}}ll f=n/2;ll ans=0;rep(j,m){for(ll i=f;i>=1;i--){if(i+x[j]>f&&i+x[j]<=n){ans+=(ll)dp[i]*dp[n-i-x[j]+1];ans%=mod;}}}//cout << ans << endl;cand.emplace_back(ans);mods.emplace_back(mod);}auto f=ChineseRem(cand,mods);cout << f.first << endl;}