#include<bits/stdc++.h>
using namespace std;
mt19937_64 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
namespace Random
{
#define int long long
int Rand(int l , int r)
{
return uniform_int_distribution<int>(l , r)(rng);
}
#undef int
}
using i64 = long long;
using ui64 = unsigned long long;
#define MASK(x) ((i64)(1) << (x))
#define BIT(mask , x) (((mask) >> (x)) & (1))
#define sz(x) (x).size()
#define all(x) (x).begin() , (x).end()
#define FOR(i ,a , b) for (int i = (a); i <= (b); ++i)
#define FORD(i , a , b) for (int i = (b); i >= (a); --i)
#define REP(i , a , b) for (int i = (a); i < (b); ++i)
#define REPD(i , a , b) for (int i = (b) - 1 ; i >= (a); --i)
template <class T>
void compress(vector<T> &a)
{
sort(a.begin() , a.end());
a.resize(unique(a.begin() , a.end()) - a.begin());
return;
}
template<class T>
void printArr(T& container , string separator = "" , string finish = "\n")
{
for (auto& item : container) cout << item << separator;
cout << finish;
}
template<class T>
bool maximize(T &a , T b) {if (a < b) return a = b , true; else return false;}
template<class T>
bool minimize(T &a , T b) {if (a > b) return a = b , true; else return false;}
template<class T>
T gcd(T x , T y) {while (y) swap(y , x %= y); return x;}
template<class T>
T lcm(T x , T y) {return (x * y) / gcd(x , y);}
//... INPUT
void INPUT(string name)
{
iostream::sync_with_stdio(false); cin.tie(0);
if (!fopen((name + ".inp").c_str() , "r")) return;
freopen((name + ".inp").c_str() , "r" , stdin);
freopen((name + ".out").c_str() , "w+" , stdout);
return;
}
const int maxn = 1e6;
int a[maxn + 2] , ID[maxn + 2] , n , k , length;
//... MOD OPERATOR
const int MOD = 1e6 + 3;
int add(int a , int b)
{
return (a + b >= MOD ? a + b - MOD : a + b);
}
int mul(int a , int b)
{
return (i64)a * b % MOD;
}
int _pow(int a , int b)
{
int res = 1;
for (; b ; b >>= 1 , a = mul(a,a))
if (b&1) res = mul(res,a);
return res;
}
int fac[maxn + 2] , inv[maxn + 2];
void pre()
{
fac[0] = 1;
FOR(i,1,maxn) fac[i] = mul(fac[i - 1] , i);
inv[maxn] = _pow(fac[maxn] , MOD - 2);
FORD(i,1,maxn) inv[i - 1] = mul(inv[i] , i);
return;
}
int C(int x , int y)
{
return (i64)fac[y] * inv[x] % MOD * inv[y - x] % MOD;
}
class segment_tree
{
public:
vector<int> st;
vector<int> lazy;
void init(int n)
{
st.assign(n*4+2,{0});
lazy.assign(n*4+2,{0});
return;
}
void build(int node , int l , int r)
{
if (l == r) st[node] = a[l];
else
{
int m = l + r >> 1;
build(node*2,l,m);
build(node*2+1,m+1,r);
st[node] = min(st[node*2],st[node*2+1]);
}
return;
}
void pushdown(int node)
{
int& t = lazy[node];
st[node*2] += t , lazy[node*2] += t;
st[node*2+1] += t , lazy[node*2+1] += t;
t = 0;
return;
}
void update(int node , int l , int r , int u , int v , int val)
{
if (l > v || r < u) return;
if (u <= l && r <= v)
{
st[node] += val;
lazy[node] += val;
return;
}
int m = l + r >> 1;
pushdown(node);
update(node*2,l,m,u,v,val);
update(node*2+1,m+1,r,u,v,val);
st[node] = min(st[node*2],st[node*2+1]);
return;
}
int get(int node , int l , int r , int u , int v)
{
if (l > v || r < u) return (int)1e9 + 7;
if (u <= l && r <= v) return st[node];
int m = l + r >> 1;
pushdown(node);
return min(get(node*2,l,m,u,v),get(node*2+1,m+1,r,u,v));
}
};
segment_tree st1;
vector<int> bag[maxn + 2];
int s[maxn + 2] , orz[maxn + 2];
void del(int x , int sub)
{
int lim = min(x + k - 1 , n);
st1.update(1,1,n,x,lim,-sub);
}
int32_t main()
{
INPUT("main");
cin >> n >> k;
pre();
length = n - k + 1;
FOR(i,k,n) cin >> a[i];
st1.init(n);
FOR(i,1,k)
{
ID[i] = i;
bag[ID[i]].push_back(i);
}
FOR(i,1,n) s[i] = -1;
st1.build(1,1,n);
FOR(i,k+1,n)
{
if (a[i] - a[i - 1]==0)
{
if (ID[i - k] == 0) s[i] = s[i - k];
else
{
ID[i] = ID[i - k];
bag[ID[i]].push_back(i);
}
}
else if (abs(a[i] - a[i - 1]) == 1)
{
if (a[i] - a[i - 1] == 1)
{
if(ID[i - k] != 0)
{
for (auto& x : bag[ID[i - k]])
{
if (s[x] == 1)
{
cout << 0;
return 0;
}
s[x] = 0;
}
bag[ID[i - k]].clear();
ID[i - k] = 0;
}
else
{
if (s[i - k] == 1)
{
cout << 0;
return 0;
}
s[i - k] = 0;
}
s[i] = 1;
del(i,s[i]);
}
else
if (a[i] - a[i - 1] == -1)
{
if(ID[i - k] != 0)
{
for (auto& x : bag[ID[i - k]])
{
if (s[x] == 0)
{
cout << 0;
return 0;
}
s[x] = 1;
del(x,s[x]);
}
bag[ID[i - k]].clear();
ID[i - k] = 0;
}
else
{
if (s[i - k] == 0)
{
cout << 0;
return 0;
}
s[i - k] = 1;
del(i-k,s[i-k]);
}
s[i] = 0;
}
}
else if (abs(a[i] - a[i - 1]) >= 2)
{
cout << 0;
return 0;
}
}
int answer = 1;
FOR(i,1,k)
{
if (s[i]==-1)
++orz[min(n,bag[i].back() + k - 1)];
}
FORD(i,k,n)
{
i64 x = st1.get(1,1,n,i,i);
if (x > 0)
{
st1.update(1,1,n,1,i,-x);
answer = mul(answer , C(x , orz[i]));
}
else if (x < 0) answer = 0;
}
cout << answer << '\n';
cerr << "\nTIME RUN : " << 1000 * clock() / CLOCKS_PER_SEC << "MS\n";
return 0;
}
Compilation message
Main.cpp: In member function 'void segment_tree::build(int, int, int)':
Main.cpp:110:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
110 | int m = l + r >> 1;
| ~~^~~
Main.cpp: In member function 'void segment_tree::update(int, int, int, int, int, int)':
Main.cpp:134:13: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
134 | int m = l + r >> 1;
| ~~^~~
Main.cpp: In member function 'int segment_tree::get(int, int, int, int, int)':
Main.cpp:145:13: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
145 | int m = l + r >> 1;
| ~~^~~
Main.cpp: In function 'void INPUT(std::string)':
Main.cpp:55:13: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
55 | freopen((name + ".inp").c_str() , "r" , stdin);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:56:13: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
56 | freopen((name + ".out").c_str() , "w+" , stdout);
| ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
15 ms |
37208 KB |
Output is correct |
| 2 |
Correct |
16 ms |
37212 KB |
Output is correct |
| 3 |
Incorrect |
16 ms |
37212 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
