UESTC OJ 1650 Electric System Restore

ac代码：

1. #include <iostream>  
2. #include <cstdio>  
3. #include <string.h>  
4. #include <algorithm>  
5.   
6. using namespace std;  
7. const int N = 1010;  
8. int num[N],px[N],py[N],visted[N];  
9. struct point  
10. {  
11.     int x,y;  
12. }pp[N];  
13.   
14. int main()  
15. {  
16.     int numcase,n,m;  
17.     scanf("%d",&numcase);  
18.     for(int k = 1;k <= numcase;++k)  
19.     {  
20.         scanf("%d%d",&n,&m);  
21.         int sum = 0;  
22.         for(int i = 1;i <= n;++i)  
23.         {  
24.             scanf("%d",&num[i]);  
25.         }  
26.        for(int i = 1;i <= n;++i)  
27.        {  
28.            scanf("%d%d",&pp[i].x,&pp[i].y);  
29.        }  
30.        bool flag = true;  
31.        memset(visted,0,sizeof(visted));  
32.        while(m && flag)  
33.        {  
34.            flag = false;  
35.            int numpoint = 0;  
36.            for(int i = 1;i <= n;++i)  
37.            {  
38.                if(!visted[i])  
39.                {  
40.                    px[numpoint] = pp[i].x;  
41.                    py[numpoint++] = pp[i].y;  
42.                }  
43.            }  
44.            sort(px,px+numpoint);  
45.            sort(py,py+numpoint);  
46.            int midx = px[(numpoint-1)/2];  
47.            int midy = py[(numpoint-1)/2];  
48.            //printf("midx= %d midy=%d\n",midx,midy);  
49.            int pos = 0,mm = 0;  
50.            for(int i = 1;i <= n;++i)  
51.            {  
52.                int y = abs(midx - pp[i].x) + abs(midy - pp[i].y);  
53.                if(!visted[i])  
54.                {  
55.                    if(num[i] <= y)  
56.                    {  
57.                        flag = true;  
58.                        if(y - num[i] > mm)  
59.                        {  
60.                            mm = y -num[i];  
61.                            pos = i;  
62.                        }  
63.                    }  
64.                }  
65.            }  
66.            if(flag)  
67.            {  
68.                visted[pos] = 1;  
69.                m--;  
70.                sum += num[pos];  
71.               // printf("pos=%d\n",pos);  
72.            }  
73.        }  
74.        int numpoint = 0;  
75.        for(int i = 1;i <= n;++i)  
76.        {  
77.            if(!visted[i])  
78.            {  
79.                px[numpoint] = pp[i].x;  
80.                py[numpoint++] = pp[i].y;  
81.            }  
82.        }  
83.        sort(px,px+numpoint);  
84.        sort(py,py+numpoint);  
85.        int midx = px[(numpoint-1)/2];  
86.        int midy = py[(numpoint-1)/2];  
87.        for(int i = 1;i <= n;++i)  
88.        {  
89.            if(!visted[i])  
90.            {  
91.                int y = abs(midx - pp[i].x) + abs(midy - pp[i].y);  
92.                sum += y;  
93.            }  
94.        }  
95.        printf("Case #%d: %d\n",k,sum);  
96.     }  
97.     return 0;  
98. }