HDU - 5156 Harry and Christmas tree

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <algorithm>
  4 #include <vector>
  5 
  6 using namespace std;
  7 
  8 const int MAXN = 5e4+10;
  9 const int MAXM = 5e5+10;
 10 
 11 vector<int> edge[MAXN];
 12 vector<int> gift[MAXN];
 13 int seq[2*MAXM];
 14 int st[MAXN];
 15 int ed[MAXN];
 16 int s[2*MAXM];
 17 int mp[2*MAXM];
 18 int ans[MAXN];
 19 int pos[MAXN];
 20 int cnt, n, m;
 21 
 22 void Dfs(int u, int fa)
 23 {
 24     st[u] = cnt + 1;
 25     int num = gift[u].size();
 26     for(int i = 0; i < num; i++) {
 27         seq[++cnt] = gift[u][i];
 28     }
 29     int len = edge[u].size();
 30     for(int i = 0; i < len; i++) {
 31         int v = edge[u][i];
 32         if(v != fa) Dfs(v, u);
 33     }
 34     ed[u] = cnt;
 35 }
 36 
 37 int Lowbit(int x)
 38 {
 39     return x & (-x);
 40 }
 41 
 42 void Add(int x, int val)
 43 {
 44     for(int i = x; i <= cnt; i += Lowbit(i)) {
 45         s[i] += val;
 46     }
 47 }
 48 
 49 int Sum(int x)
 50 {
 51     int res = 0;
 52     for(int i = x; i > 0; i -= Lowbit(i)) {
 53         res += s[i];
 54     }
 55     return res;
 56 }
 57 
 58 bool cmp(int a, int b)
 59 {
 60     return ed[a] < ed[b];
 61 }
 62 
 63 void Init()
 64 {
 65     for(int i = 0; i <= n; i++) {
 66         edge[i].clear();
 67         gift[i].clear();
 68     }
 69     for(int i = 1; i <= n; i++) {
 70         pos[i] = i;
 71     }
 72     memset(s, 0, sizeof(s));
 73     memset(mp, 0, sizeof(mp));
 74 }
 75 
 76 int main()
 77 {
 78     int u, v;
 79 
 80     while(scanf("%d %d", &n, &m) != EOF) {
 81         Init();
 82         for(int i = 0; i < n-1; i++) {
 83             scanf("%d %d", &u, &v);
 84             edge[u].push_back(v);
 85             edge[v].push_back(u);
 86         }
 87         while(m--) {
 88             scanf("%d %d", &u, &v);
 89             if(find(gift[u].begin(), gift[u].end(), v) == gift[u].end()) {
 90                 gift[u].push_back(v);
 91             }
 92         }
 93         cnt = 0;
 94         Dfs(1, -1);
 95         sort(pos+1, pos+1+n, cmp);
 96         for(int i = 1; i <= cnt; i++) {
 97             if(mp[seq[i]] == 0) {
 98                 Add(i, 1);
 99                 mp[seq[i]] = i; //如果是第一次出现, mp[seq[i]]记录为当前位置
100             }
101         }
102         int right = 1;
103         for(int i = 1; i <= n; i++) {
104             int now = pos[i];
105             while(right <= ed[now]) {
106                 if(mp[seq[right]] != right) { //如果不是第一次出现
107                     Add(mp[seq[right]], -1);  //减去前一次出现的
108                     Add(right, 1);
109                     mp[seq[right]] = right;   //重新定义这个数最近一次出现位置
110                 }
111                 right++;
112             }
113             ans[now] = Sum(ed[now]) - Sum(st[now] - 1);
114         }
115         printf("%d", ans[1]);
116         for(int i = 2; i <= n; i++) {
117             printf(" %d", ans[i]);
118         }
119         printf("\n");
120     }
121 
122     return 0;
123 }

1 for(int i = 1; i <= n; i++) {
2     if(mp[a[i]] == 0) { //如果是第一次出现, 记录第一次出现位置并更新
3         Add(i, 1);
4         mp[a[i]] = i;
5     }
6 }

 1 for(int i = 1; i <= query_num; i++) {
 2     while(k <= R[i]) {
 3         if(mp[a[i]] != k) {     //如果不是最新次出现
 4             Add(mp[a[k]], -1);  //将上一次出现的更新-1
 5             Add(k, 1);　　       //将这个位置新出现的更新1
 6             mp[a[k]] = k;       //更新这个数最近一次的位置
 7         }
 8         k++;
 9     }
10     ans[i] = Sum(ed[i]) - Sum(st[i] - 1);
11 }