A remote island chain contains n islands, labeled 1 through n. Bidirectional bridges connect the islands to form a simple cycle — a bridge connects islands 1 and 2, islands 2 and 3, and so on, and additionally a bridge connects islands n and 1. The center of each island contains an identical pedestal, and all but one of the islands has a fragile, uniquely colored statue currently held on the pedestal. The remaining island holds only an empty pedestal.
The islanders want to rearrange the statues in a new order. To do this, they repeat the following process: First, they choose an island directly adjacent to the island containing an empty pedestal. Then, they painstakingly carry the statue on this island across the adjoining bridge and place it on the empty pedestal.
Determine if it is possible for the islanders to arrange the statues in the desired order.
The first line contains a single integer n (2 ≤ n ≤ 200 000) — the total number of islands.
The second line contains n space-separated integers ai (0 ≤ ai ≤ n - 1) — the statue currently placed on the i-th island. If ai = 0, then the island has no statue. It is guaranteed that the ai are distinct.
The third line contains n space-separated integers bi (0 ≤ bi ≤ n - 1) — the desired statues of the ith island. Once again, bi = 0indicates the island desires no statue. It is guaranteed that the bi are distinct.
Print "YES" (without quotes) if the rearrangement can be done in the existing network, and "NO" otherwise.
3
1 0 2
2 0 1
YES
2
1 0
0 1
YES
4
1 2 3 0
0 3 2 1
NO
In the first sample, the islanders can first move statue 1 from island 1 to island 2, then move statue 2 from island 3 to island 1, and finally move statue 1 from island 2 to island 3.
In the second sample, the islanders can simply move statue 1 from island 1 to island 2.
In the third sample, no sequence of movements results in the desired position.
思路:KMP
其实0可以省略看成没有,去0后不改变原来的顺序,其实只要适当移动0可以移到任意位置,所以把0全移到最前这样就能看成没有02020---00022。然后把去0后的原窜复制一遍放在后面,然后下面的窜去0后对上串KMP即可。O(n);
1 #include<stdio.h> 2 #include<algorithm> 3 #include<iostream> 4 #include<string.h> 5 #include<math.h> 6 #include<queue> 7 #include<map> 8 using namespace std; 9 typedef long long LL; 10 void next(int n); 11 int aa[200005]; 12 int a[400005]; 13 int bb[200005]; 14 int f[200005]; 15 int nex[200005]; 16 int main(void) 17 { 18 int i,j,k,p,q; 19 cin>>k;int cnt=0; 20 for(i=0;i<k;i++) 21 { 22 cin>>p; 23 if(p!=0) 24 {cnt++;a[cnt]=aa[cnt]=p;} 25 } 26 for(i=cnt+1;i<=2*cnt;i++) 27 { 28 a[i]=aa[i-cnt]; 29 } 30 int ans=0; 31 for(i=0;i<k;i++) 32 { 33 cin>>p; 34 if(p!=0) 35 { ans++; 36 bb[ans]=p; 37 38 } 39 } 40 next(ans); 41 j=0;int flag=0; 42 for(i=1;i<=2*cnt;i++) 43 { 44 while(j>0&&bb[j+1]!=a[i]) 45 { 46 j=nex[j]; 47 } 48 if(bb[j+1]==a[i]) 49 { 50 j++; 51 } 52 if(j==ans) 53 {flag=1; 54 break; 55 } 56 }if(flag) 57 { 58 printf("YES "); 59 } 60 else printf("NO "); 61 return 0; 62 } 63 void next(int n) 64 { 65 int i,j; 66 nex[0]=0; 67 nex[1]=0;j=0; 68 for(i=2;i<=n;i++) 69 { 70 while(bb[j+1]!=bb[i]&&j>0) 71 { 72 j=nex[j]; 73 } 74 if(bb[j+1]==bb[i]) 75 { 76 j++; 77 } 78 nex[i]=j; 79 } 80 }