Codeforces Round #595（Div.3.A B1 B2 C）

A

这道题的输出只有1或者2
如果没有绝对值小于1的一队，那么输出1
如果有，那么可以分成2个队伍
代码如下：

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
 int q;
 cin>>q;
 for(int i=0;i<q;i++){
  int n;
  cin>>n;
  int a[105];
  for(int j=0;j<n;j++){
   cin>>a[j];
  }
  if(n>1){
  sort(a,a+n);//可以不排序，直接两个for循环嵌套
  int flag=1;
  for(int k=0;k<n-1;k++){
   if(a[k+1]-a[k]==1){
    flag++;
    break;
   }
  }
  cout<<flag<<endl;
 }
 else cout<<1<<endl;
 }
 return 0;
 }

---

B1

这道题很简单，就是不断地检索
代码如下：

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
 int q;
 cin>>q;
 for(int i=1;i<=q;i++){
  int n;
  cin>>n;
  int a[205];
  for(int j=1;j<=n;j++){
   cin>>a[j];
  }
  for(int j=1;j<=n;j++){
   bool flag=false;
   int t=a[j];
   int sum=1;
   	while(t!=j){
    	sum++;
    	t=a[t];
   	}
   cout<<sum<<' ';
  }
  cout<<endl;
 }
 return 0;
 } 

---

B2

和B1的区别就是数据从200扩大到2*10⁵
相应地，我们也要优化算法避免超时

下面转载了官方给出的伪代码
时间复杂度为O（n）：

pos = p[i]
ans = 1
cycle = [i]
while pos != i:
    cycle.append(pos)
    ans += 1
    pos = p[pos]
for el in cycle: res[el] = ans

---

C

待更新