oj.uz

답안 #1026146

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1026146 2024-07-17T15:51:00 Z _rain_ Binaria (CCO23_day1problem1) C++14
0 / 25
 4 ms 8792 KB 
#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 = 2e5;
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)
	{
		if (x > y) return 0;
		return 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);
			}
		st1.build(1,1,n);
		FOR(i,k+1,n)
		{
			a[i] = st1.get(1,1,n,i,i);
			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]]) 
							s[x] = 0;
						bag[ID[i - k]].clear();
						ID[i - k] = 0;
					}
					else 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]]) 
							{
								s[x] = 1;
								del(x,s[x]);
							}
						
						bag[ID[i - k]].clear();
						ID[i - k] = 0;
					}
					else 
					{
						s[i - k] = 1;
						del(i-k,s[i-k]);
					}
					s[i] = 0;
				}
			}
			else 
			{
				cout << 0;
				return 0;
			}
		}
		int answer = 1;
		FOR(i,1,k)
		{
			if (sz(bag[i])) 
					orz[min(n,bag[i].back()+k-1)]++;
		}
		FORD(i,length,n)
			{
				int x = st1.get(1,1,n,i,i);
				if (orz[i] && x > 0)
				{
					answer = mul(answer , C(x , orz[i]));
					st1.update(1,1,n,length,i,-x);
				}
			}
		
		FOR(i,length,n) if (st1.get(1,1,n,i,i) != 0)
			{
				cout << 0 << '\n';
				return 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:111:14: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  111 |    int m = l + r >> 1;
      |            ~~^~~
Main.cpp: In member function 'void segment_tree::update(int, int, int, int, int, int)':
Main.cpp:135:13: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  135 |   int m = l + r >> 1;
      |           ~~^~~
Main.cpp: In member function 'int segment_tree::get(int, int, int, int, int)':
Main.cpp:146:13: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
  146 |   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);
      |      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 3.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #2
0 / 3.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #4
0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #6
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 8792 KB Output is correct
2 Correct 4 ms 8540 KB Output is correct
3 Incorrect 3 ms 8668 KB Output isn't correct
4 Halted 0 ms 0 KB -