HDU 4446 IT Companies (线段树, 区间修改)

题目链接: HDU 4446

• 题意: 有n个公司，标签由1到n, 对应的分支标签由-1到-n. （分支不是公司，分支和公司均是单元）
• 满足以下条件:
  
  • 每个公司的人数>对应分支的人数
  • 如果A公司人数>B公司人数，那么A公司的分支的人数>B公司的分支的人数
• 现在给出C[i], 表示对于第i个公司而言，在标签>i (即i+1到n) 的公司 和 标签大于-i (即 -(i-1) 到-1)的分支中有C[i]个单元的人数大于i公司.求公司与分支的顺序 (按照人数从小到大输出标签)

• 
  
• 官方题解:
  

1．一开始开2个队列a,b。a为答案队列，b为临时队列
2．找出当最小的C值为C[k]（相同时取k最小的）
a)C[k]=0：
将k加入a,-k加入b. 在C序列中删除元素C[k]，并将C[1]到C[k-1]减一。
b)C[k]>0：将b队首的元素p取出加入a，并将C[-p+1]到c[n]减一。
重复做直到所有元素进入a或者b
3．将b接在a后面，倒序就是答案序列
无解有两种情况：
1．C中出现了负数。
2．出现2b情况时b为空

/*
线段树求解，区间更新 
*/
#include<bits/stdc++.h>
#define rep(i, a, b) for(int i=a; i<=b; ++i)
#define repp(i, a, b) for(int i=b; i>=a; --i)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define ms(a, b) memset(a, b, sizeof(a))
#define de(a) cout <<#a << " => "<<a<<endl;
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef pair<int, int> pii;
const int maxn = 1e5+7, inf = 1e8;
//-----------------------------------------------
pii st[maxn << 2];
int lazy[maxn << 2];
void Update(int rt){
    if(st[rt << 1].fi <= st[rt << 1 | 1].fi){
        st[rt] = st[rt << 1];
    } else {
        st[rt] = st[rt<<1 | 1];
    }
    st[rt].fi = st[rt].fi + lazy[rt];
    return ;
}
void build(int rt, int l, int r){
    lazy[rt] = 0;
    if(l == r){
        scanf("%d", &st[rt].fi);
        st[rt].se = l;
    } else {
        int m = (l + r) >> 1;
        build(rt << 1, l, m);
        build(rt << 1 | 1, m+1, r);
        Update(rt);
    }
    return ;
}
void add(int rt, int l , int r, int L, int R, int ad){
    if( L > R) return ;
    if(L <= l && r <= R){
        st[rt].fi += ad;
        lazy[rt] += ad;
    } else {
        int m = (l+r)>>1;
        if(L <= m) add(rt << 1, l, m, L, R, ad);
        if(R > m) add(rt << 1 | 1, m+1, r, L, R, ad);
        Update(rt);
    }
    return ;
}
int n;
queue<int>ans, tmp;
int main(){
    while(~scanf("%d", &n), n){
        build(1, 1, n);
        while(!ans.empty()) ans.pop();
        while(!tmp.empty()) tmp.pop();
        bool flag=1;
        int cnt=2*n;
        while(ans.size() + tmp.size() < cnt){
            pii a = st[1];
            if(a.fi < 0) {
                flag = 0;
                break;
            } else if(a.fi == 0){
                ans.push(a.se);
                tmp.push(-a.se);
                add(1, 1, n, 1, a.se-1, -1);
                add(1, 1, n, a.se, a.se, inf);
            } else {
                if(tmp.empty()) {
                    flag=false;
                    break;
                }
                int s = tmp.front();
                tmp.pop();
                ans.push(s);
                add(1, 1, n, -s+1, n, -1);  
            }
        }
        if(!flag){
            cout <<"Impossible";
        } else{
            while(!tmp.empty()) {
                ans.push(tmp.front());
                tmp.pop();
            }
            stack<int>sk;
            while(!ans.empty()){
                sk.push(ans.front());
                ans.pop();  
            }
            while(!sk.empty()){
                printf("%d ", sk.top());
                sk.pop();
            }   
        }
        puts(""); 
    }
    return 0;
}