[Codeforces Round #679 (Div. 2)] D. Shurikens (思維,樹狀陣列)
阿新 • • 發佈:2020-10-26
[Codeforces Round #679 (Div. 2)] D. Shurikens (思維,樹狀陣列)
題面:
題意:
現在有價格分別為\([1,n]\)的武器,以及2種操作。
- +意味著店主拿一個武器放在展臺;
- -x意味客戶買了價格為\(\mathit x\)的武器。
初始時展臺為空,且客戶們總是買當前展臺裡最便宜的武器。
問你是否有一個放武器的順序滿足給定的操作。
思路:
我們只需要維護出當前放入展臺的武器的價格的最大值和最小值即可,
維護方法:
1、連續的購物武器時,後買的價格一定比前面的高。
2、用樹狀陣列單點修改區間查詢和當前展臺中武器的個數可以得出當前購買武器時的價格上限。
3、如果當前展臺中武器個數大於1,那麼這次購買武器的價格為\(num\)時,展臺中剩餘武器的價格一定大於\(num\),可以得出價格下限。
程式碼:
#include <iostream> #include <cstdio> #include <algorithm> #include <bits/stdc++.h> #define ALL(x) (x).begin(), (x).end() #define sz(a) int(a.size()) #define rep(i,x,n) for(int i=x;i<n;i++) #define repd(i,x,n) for(int i=x;i<=n;i++) #define pii pair<int,int> #define pll pair<long long ,long long> #define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0) #define MS0(X) memset((X), 0, sizeof((X))) #define MSC0(X) memset((X), '\0', sizeof((X))) #define pb push_back #define mp make_pair #define fi first #define se second #define eps 1e-6 #define chu(x) if(DEBUG_Switch) cout<<"["<<#x<<" "<<(x)<<"]"<<endl #define du3(a,b,c) scanf("%d %d %d",&(a),&(b),&(c)) #define du2(a,b) scanf("%d %d",&(a),&(b)) #define du1(a) scanf("%d",&(a)); using namespace std; typedef long long ll; ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;} ll lcm(ll a, ll b) {return a / gcd(a, b) * b;} ll powmod(ll a, ll b, ll MOD) { if (a == 0ll) {return 0ll;} a %= MOD; ll ans = 1; while (b) {if (b & 1) {ans = ans * a % MOD;} a = a * a % MOD; b >>= 1;} return ans;} ll poww(ll a, ll b) { if (a == 0ll) {return 0ll;} ll ans = 1; while (b) {if (b & 1) {ans = ans * a ;} a = a * a ; b >>= 1;} return ans;} void Pv(const vector<int> &V) {int Len = sz(V); for (int i = 0; i < Len; ++i) {printf("%d", V[i] ); if (i != Len - 1) {printf(" ");} else {printf("\n");}}} void Pvl(const vector<ll> &V) {int Len = sz(V); for (int i = 0; i < Len; ++i) {printf("%lld", V[i] ); if (i != Len - 1) {printf(" ");} else {printf("\n");}}} inline long long readll() {long long tmp = 0, fh = 1; char c = getchar(); while (c < '0' || c > '9') {if (c == '-') { fh = -1; } c = getchar();} while (c >= '0' && c <= '9') { tmp = tmp * 10 + c - 48, c = getchar(); } return tmp * fh;} inline int readint() {int tmp = 0, fh = 1; char c = getchar(); while (c < '0' || c > '9') {if (c == '-') { fh = -1; } c = getchar();} while (c >= '0' && c <= '9') { tmp = tmp * 10 + c - 48, c = getchar(); } return tmp * fh;} void pvarr_int(int *arr, int n, int strat = 1) {if (strat == 0) {n--;} repd(i, strat, n) {printf("%d%c", arr[i], i == n ? '\n' : ' ');}} void pvarr_LL(ll *arr, int n, int strat = 1) {if (strat == 0) {n--;} repd(i, strat, n) {printf("%lld%c", arr[i], i == n ? '\n' : ' ');}} const int maxn = 1000010; const int inf = 0x3f3f3f3f; /*** TEMPLATE CODE * * STARTS HERE ***/ #define DEBUG_Switch 0 int n; int ans[maxn]; int isok = 1; stack<int> st; int tree[maxn]; int lowbit(int x) { return (-x)& x; } void add(int pos, int val) { while (pos < maxn) { tree[pos] += val; pos += lowbit(pos); } } int ask(int pos) { int res = 0; while (pos > 0) { res += tree[pos]; pos -= lowbit(pos); } return res; } int pre[maxn]; int main() { #if DEBUG_Switch freopen("C:\\code\\input.txt", "r", stdin); #endif //freopen("C:\\code\\output.txt","w",stdout); cin >> n ; int id = 0; int mi = 0; int cnt = 0; repd(i, 1, n + n) { char op; cin >> op; if (op == '+') { st.push(++id); mi = 0; } else { int x; cin >> x; cnt = ask(n) - ask(x - 1); if (isok == 0 || x < mi || 0 == (int)st.size() || x > n - sz(st) + 1 - cnt || x < pre[st.top()] ) { isok = 0; continue; } mi = x; ans[st.top()] = x; st.pop(); if (st.size() > 0) { pre[st.top()] = max(pre[st.top()], x + 1); } add(x, 1); } } if (isok) { printf("YES\n"); repd(i, 1, n) { printf("%d%c", ans[i], i == n ? '\n' : ' ' ); } } else { printf("NO\n"); } return 0; }