Code 備忘錄►Else

STEP5::Problem 0120 : 蘿莉交換問題

http://web2.ck.tp.edu.tw/~step5/probdisp.php?pid=0120
題目要交換兩行陣列中對應的數字，使得一陣列中數字不重複。
方法是先將兩個陣列並排排好，兩個兩個連起來，再把一樣的數字連起來，會連成很多環或是鍊，在任意一條上依序標上010101….

標完後觀察其中一個陣列中0或1的數量，較少的便是需要交換的次數(每一條鍊或環是分開處理的)，把所有鍊或環的交換次數加總，結束~

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;

vector<int> lin[100001],A,B;
queue<int>  que;

int n,loli[50001][2];
bool loli2[50001]={0},br;

int link(int x,int t,int a,bool br)
{
    if(br)return (A[t]<B[t])? A[t]:B[t];
    else loli2[x]=1;
    
    if(a==0)A[t]+=1;
    else B[t]+=1;
    //printf("a:%d\n",a);
    int p=loli[x][(a+1)%2],p2=0;
    if(lin[p][p2]==x)p2+=2;
    
    if(lin[p][p2]==-1)return (A[t]<B[t])? A[t]:B[t];
    
    return link(lin[p][p2],t,lin[p][p2+1],loli2[lin[p][p2]]);
}

int fin(int x,int t,int a,bool br,bool *l)
{
    if(br)return 0;
    else loli2[x]=1;
    
    int p=loli[x][(a+1)%2],p2=0,c;
    if(lin[p][p2]==x)p2+=2;
    
    if(lin[p][p2]==-1){
                loli2[x]=0;
                *l=(a+1)%2;
                return x;
    }    
    c=fin(lin[p][p2],t,lin[p][p2+1],loli2[lin[p][p2]],l);
    loli2[x]=0;
    if(c) return c;
    else return 0; 
}


int main(int argc,char *argv[])
{
    
    scanf("%d",&n);
    for(int i=1;i<n+1;i++){
            int tmp;
            scanf("%d",&tmp);
            loli[i][0]=tmp;
            lin[tmp].push_back(i);
            lin[tmp].push_back(0);
            if(lin[tmp].size()==2)que.push(tmp);
    }
    for(int i=1;i<n+1;i++){
            int tmp;
            scanf("%d",&tmp);
            loli[i][1]=tmp;
            lin[tmp].push_back(i);
            lin[tmp].push_back(1);
            if(lin[tmp].size()==2)que.push(tmp);
    }
    
    for(int i=que.front();!que.empty();i=que.front()){
            if(lin[i].size()==2){
                    lin[i].push_back(-1);
                    lin[i].push_back(-1);
            }
            que.pop();
    }
    int t=0,sum=0,c;
    bool l;
    for(int i=1;i<n+1;i++)
            if(!loli2[i]){
                    A.push_back(0);
                    B.push_back(0);
                    c=fin(i,t,0,0,&l);
                    if(c) sum+=link(c,t,l,0);
                    else sum+=link(i,t,0,0);
                    t++;
                    //link(i,t++,0,0);
            }
    //printf("AB:%d %d\n",A[0],B[0]);
    //printf("AB:%d %d\n",A[1],B[1]);
           
    printf("%d\n",sum);
    
    //system("pause");
    return EXIT_SUCCESS;
}