URAL--1827 Indigenous Wars（区间）

求区间覆盖。

http://acm.timus.ru/problem.aspx?space=1&num=1827

从别人那里学来的方法。

判定区间[i, j]是否被覆盖。首先开一个数组add，初始化为0.若[i, j]覆盖，将add[i - 1]-- , add[j]++

处理完之后，从后向前扫一遍，add[i] += add[i + 1].如果add[i]>0，则i被覆盖。

例如区间[2, 5]覆盖，处理完后add数组为0 -1 0 0 0 1

扫一遍之后就变成了0 0 1 1 1 1

#include <map>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

#define pb push_back
#define MP make_pair

typedef long long ll;
typedef pair<int, int> PII;

const int inf = 0x3f3f3f3f;
const int mod = 1000000007;
const int maxn = 100000 + 10;

int n, m, a[maxn], add[maxn];
struct node{
    int s, t, len;
    bool operator < (const node& x) const {
        return (s < x.s) || (s == x.s && len > x.len);
    }
}rec[maxn];
map<int, int> vis, tmp;
int main(){
    while(scanf("%d", &n) == 1){
        vis.clear(); tmp.clear();
        memset(add, 0, sizeof(add));
        for(int i=n; i>=1; i--){
            scanf("%d", &a[i]);
            vis[a[i]] = 1;
        }
        scanf("%d", &m);
        int cnt = 0;
        for(int i=0; i<m; i++){
            scanf("%d%d%d", &rec[cnt].s, &rec[cnt].t, &rec[cnt].len);
            if(!vis[rec[cnt].s] || !vis[rec[cnt].t] || rec[cnt].len > n) cnt--;
            cnt++;
        }
        sort(rec, rec + cnt);
        tmp[rec[0].s] = 1;
        for(int i=1; i<cnt; i++){
            if(rec[i].s != rec[i - 1].s)
                tmp[rec[i].s] = i + 1;
        }
        for(int i=1; i<=n; i++){
            if(!tmp[a[i]]) continue;
            for(int j=tmp[a[i]]-1; j<cnt; j++){
                if(rec[j].s != a[i])break;
                if(i + rec[j].len - 1 > n) continue;
                if(rec[j].t == a[i + rec[j].len - 1]){
                    add[i - 1]--;
                    add[i + rec[j].len - 1]++;
                    break;
                }
            }
        }
        for(int i=n; i>=1; i--){
            add[i] += add[i + 1];
            if(add[i]) printf("1");
            else printf("0");
        }
        puts("");
    }
    return 0;
}