洛谷mNOIP模拟赛Day1-分组

传送门

---

首先是贪心的思路

从后向前选，能多选就多选，

理由：数字越少肯定越优，同时间隔尽量向前推，字典序尽量小

对于K==1，枚举1~512直接判断

对于K==2，需要用镜像并查集，来刻画“敌对关系”，如果a和b产生矛盾，就把a和b的镜像（b'）连接 ，b和a'连接，然后判断自己是不是和自己的镜像连接了

打上时间戳避免清零卡常

 #include<cstdio>
 #include<cstdlib>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #define MAXN 131100
 using namespace std;
 int n,K;
 namespace solve1
 {
     int read(){
         int x=;char ch=getchar();
         while(ch<''||ch>''){ch=getchar();}
         while(ch>=''&&ch<=''){x=x*+(ch^);ch=getchar();}
         return x;
     }
     int n,K,cnt;
     int a[MAXN];
     int b[MAXN<<];
     int p[];
     vector<int> v;
     int check(int x){
         for(int i=;i>=;i--){
             if(p[i]-x<=){
                 return ;
             }
             if(b[p[i]-x]&&b[p[i]-x]==cnt){
                 return ;
             }
         }
         return ;
     }
     void solve(){
         n=::n,K=::K;
         cnt=;
         for(int i=;i<=;i++){
             p[i]=i*i;
         }
         for(int i=;i<=n;i++){
             a[i]=read();
         }
         for(int i=n;i>=;i--){
             if(check(a[i])){
                 b[a[i]]=cnt;
             }
             else{
                 cnt++;
                 b[a[i]]=cnt;
                 v.push_back(i);
             }
         }
         printf("%d\n",cnt);
         for(int i=v.size()-;i>=;i--){
             printf("%d ",v[i]);
         }
     }
 }
 namespace solve2
 {
     int read(){
         int x=;char ch=getchar();
         while(ch<''||ch>''){ch=getchar();}
         while(ch>=''&&ch<=''){x=x*+(ch^);ch=getchar();}
         return x;
     }
     int f[MAXN<<];
     vector<int> v[MAXN<<];
     int vis[MAXN<<];
     int p[];
     int a[MAXN];
     int n,K;
     int cnt;
     vector<int> ans;
     int find(int x){
         return (f[x]==x?x:f[x]=find(f[x]));
     }
     void lik(int x,int y){
         x=find(x),y=find(y);
         if(x!=y){
             f[x]=y;
         }
     }
     bool same(int x,int y){
         return (find(x)==find(y));
     }
     int check(int x){
         for(int i=;i>=;i--){
             if(p[i]-a[x]<=){
                 return ;
             }
             else if(vis[p[i]-a[x]]&&vis[p[i]-a[x]]==cnt){
                 for(int j=;j<v[p[i]-a[x]].size();j++){
                     int t=v[p[i]-a[x]][j];
                     if(same(x,t)){
                         return ;
                     }
                     lik(x,t+n);
                     lik(x+n,t);
                 }
             }
         }
         return ;
     }
     void solve(){
         n=::n,K=::K;
         cnt=;
         for(int i=;i<=;i++){
             p[i]=i*i;
         }
         for(int i=;i<=(n<<);i++){
             f[i]=i;
         }
         for(int i=;i<=n;i++){
             a[i]=read();
         }
         for(int i=n;i>=;i--){
             if(check(i)){
                 if(vis[a[i]]!=cnt){
                     vis[a[i]]=cnt;
                     v[a[i]].clear();
                 }
                 v[a[i]].push_back(i);
             }
             else{
                 cnt++;
                 vis[a[i]]=cnt;
                 v[a[i]].clear();
                 v[a[i]].push_back(i);
                 ans.push_back(i);
             }
         }
         printf("%d\n",cnt);
         for(int i=ans.size()-;i>=;i--){
             printf("%d ",ans[i]);
         }
     }
 }
 int main()
 {
 //    freopen("data.in","r",stdin);
 //    freopen("a.out","w",stdout);
     scanf("%d%d",&n,&K);
     if(==K){
         solve1::solve();
     }
     else{
         solve2::solve();
     }
     return ;
 }