SRM435 Div2 Medium

500

やるだけ。

view plainprint?

1. class CellRemoval {  
2. public:  
3.   int cellsLeft(vector <int> p, int del)  
4.   {  
5.     if (p[del] == -1) return 0;  
6.   
7.     const int size = p.size();  
8.     int cnt[size];  
9.     fill(cnt, cnt + size, 0);  
10.   
11.     for (int i = 0; i < size; ++i) {  
12.       for (int n = i; p[n] != -1; n = p[n]) {  
13.         ++cnt[n];  
14.       }  
15.     }  
16.   
17.     int ret = count(cnt, cnt + size, 1);  
18.     for (int i = 0; i < size; ++i) {  
19.       if (cnt[i] != 1) continue;  
20.       for (int n = i; p[n] != -1; n = p[n]) {  
21.         if (n == del) {  
22.           --ret;  
23.           break;  
24.         }  
25.       }  
26.     }  
27.   
28.     return ret;  
29.   }  
30. };  

class CellRemoval {
public:
  int cellsLeft(vector <int> p, int del)
  {
    if (p[del] == -1) return 0;

    const int size = p.size();
    int cnt[size];
    fill(cnt, cnt + size, 0);

    for (int i = 0; i < size; ++i) {
      for (int n = i; p[n] != -1; n = p[n]) {
        ++cnt[n];
      }
    }

    int ret = count(cnt, cnt + size, 1);
    for (int i = 0; i < size; ++i) {
      if (cnt[i] != 1) continue;
      for (int n = i; p[n] != -1; n = p[n]) {
        if (n == del) {
          --ret;
          break;
        }
      }
    }

    return ret;
  }
};