【模拟8.03】分组

好题

k=1做法：

直接倒着找，满足贪心性质，预处理出每个平方数就行.....

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<string>
 4 #include<algorithm>
 5 #include<cmath>
 6 #include<vector>
 7 #include<map>
 8 #include<set>
 9 #include<cstring>
10 #define MAXN 300001
11 #define int long long
12 #define ps push_back
13 using namespace std;
14 int a[MAXN];
15 int n,K,kuan;
16 bool biao[MAXN];
17 int ans[MAXN];
18 int poww[MAXN];
19 void work1()
20 {
21     int end=n;
22     while(end>0)
23     {
24         //printf("end=%lld\n",end);
25         if(biao[a[end]]==1)
26         {
27             memset(biao,0,sizeof(biao));
28             ans[++ans[0]]=end;
29             for(int i=2;i<=sqrt(kuan)+1;++i)
30             {
31                  if(a[end]>poww[i])continue;
32                  biao[poww[i]-a[end]]=1;
33                  //printf("1::biao[%lld]\n",poww[i]-a[end]);
34             }
35             end--;
36         }
37         else
38         {
39             for(int i=2;i<=sqrt(kuan)+1;++i)
40             {
41                  if(a[end]>poww[i])continue;
42                  biao[poww[i]-a[end]]=1;
43                  //printf("biao[%lld]\n",poww[i]-a[end]);
44             }
45             end--;
46         }
47     }
48     printf("%lld\n",ans[0]+1);
49     if(ans[0]==0)printf("\n");
50     for(int i=ans[0];i>=1;--i)
51     {
52         printf("%lld ",ans[i]);
53     }
54     cout<<endl;
55 }
56 signed main()
57 {
58     //freopen("text.in","r",stdin);
59     //freopen("wa.out","w",stdout);   
60     scanf("%lld%lld",&n,&K);
61     int maxn;
62     for(int i=1;i<=n;++i)
63     {
64         scanf("%lld",&a[i]);
65         maxn=max(maxn,a[i]);
66     }
67     kuan=maxn+maxn;
68     for(int i=2;i<=sqrt(kuan)+1;++i)
69     {
70         poww[i]=i*i;
71     }
72     if(K==1)work1();
73     else printf("1\n\n");
74 } 
75 /*
76 5 1
77 1 3 15 10 6
78 
79 */
k==1做法

k=2做法：

其实这个我没打出正解并查集

用二分图强行卡常A了，把每个敌人和自己连边，染色判断

 

  1 #include<bits/stdc++.h>
  2 #include<iostream>
  3 #include<cstdio>
  4 #include<string>
  5 #include<algorithm>
  6 #include<cmath>
  7 #include<vector>
  8 #include<map>
  9 #include<set>
 10 #include<cstring>
 11 #define MAXN 1000001
 12 #define ps push_back
 13 using namespace std;
 14 int a[MAXN];
 15 vector<int>v[650001];
 16 int n,K;bool tong[MAXN];
 17 bool biao[MAXN];
 18 int ans[MAXN];
 19 int poww[MAXN];
 20 int last[MAXN];
 21 void work1()
 22 {
 23         int p=n;
 24         for(int i=n;i>=1;i--)
 25         {
 26             for(int j=512;j>=1;j--)
 27             {
 28                 if(poww[j]<=a[i]) break;
 29                 if(last[poww[j]-a[i]])
 30                 {
 31                     for(int k=p;k>i;k--) 
 32                         last[a[k]]=0;
 33                     ans[++ans[0]]=i,p=i;
 34                     break;
 35                 }
 36             }
 37             last[a[i]]++;
 38         }
 39         printf("%d\n",ans[0]+1);
 40         for(int i=ans[0];i>=1;i--) 
 41             printf("%d ",ans[i]);
 42         cout<<endl;
 43 }
 44 int col[MAXN];
 45 bool DFS(int x,int color,int r,int l,int fa)
 46 {
 47     col[x]=color;
 48     for(int i=0;i<v[x].size();i++)
 49     {
 50         int to=v[x][i];
 51         if(to==fa||to>r||to<l)continue;
 52         if(col[to]==0)
 53         {
 54             if(!DFS(to,3-color,r,l,x))
 55             return 0;
 56         }
 57         if(col[to]==col[x])
 58         {
 59              return 0;
 60         }
 61     }
 62     return 1;
 63 }
 64 signed main()
 65 { 
 66     scanf("%d%d",&n,&K);
 67     for(int i=1;i<=n;++i)
 68     {
 69         scanf("%d",&a[i]);
 70     }  
 71     for(int i=1;i<=522;++i)
 72     {
 73         poww[i]=i*i;
 74         tong[i*i]=1;
 75     }
 76     if(K==1)work1();
 77     else 
 78     {
 79       int tail=n;
 80       for(int end=n;end>=1;--end)
 81       {
 82          for(int i=tail;i>end;i--)
 83          {
 84               if(tong[a[i]+a[end]])
 85               {
 86                  v[i].ps(end);v[end].ps(i);
 87               }
 88          }
 89          if(!DFS(end,1,tail,end,0))
 90          {
 91               ans[0]++;
 92               ans[ans[0]]=end;
 93               tail=end;
 94               v[end].clear();
 95          }
 96          for(int i=tail;i>=end;i--)
 97          {
 98              col[i]=0;
 99          }
100       }
101       printf("%d\n",ans[0]+1);
102       if(ans[0]==0)cout<<endl;
103       for(int i=ans[0];i>=1;--i)
104       {
105           printf("%d ",ans[i]);
106       }   
107     }   
108 }
View Code

 

posted @ 2019-08-04 20:37  Wwb_star  阅读(188)  评论(0)  编辑  收藏  举报