【模拟8.03】分组

好题

k=1做法：

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

#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
#include<set>
#include<cstring>
#define MAXN 300001
#define int long long
#define ps push_back
using namespace std;
int a[MAXN];
int n,K,kuan;
bool biao[MAXN];
int ans[MAXN];
int poww[MAXN];
void work1()
{
    int end=n;
    while(end>0)
    {
        //printf("end=%lld
",end);
        if(biao[a[end]]==1)
        {
            memset(biao,0,sizeof(biao));
            ans[++ans[0]]=end;
            for(int i=2;i<=sqrt(kuan)+1;++i)
            {
                 if(a[end]>poww[i])continue;
                 biao[poww[i]-a[end]]=1;
                 //printf("1::biao[%lld]
",poww[i]-a[end]);
            }
            end--;
        }
        else
        {
            for(int i=2;i<=sqrt(kuan)+1;++i)
            {
                 if(a[end]>poww[i])continue;
                 biao[poww[i]-a[end]]=1;
                 //printf("biao[%lld]
",poww[i]-a[end]);
            }
            end--;
        }
    }
    printf("%lld
",ans[0]+1);
    if(ans[0]==0)printf("
");
    for(int i=ans[0];i>=1;--i)
    {
        printf("%lld ",ans[i]);
    }
    cout<<endl;
}
signed main()
{
    //freopen("text.in","r",stdin);
    //freopen("wa.out","w",stdout);   
    scanf("%lld%lld",&n,&K);
    int maxn;
    for(int i=1;i<=n;++i)
    {
        scanf("%lld",&a[i]);
        maxn=max(maxn,a[i]);
    }
    kuan=maxn+maxn;
    for(int i=2;i<=sqrt(kuan)+1;++i)
    {
        poww[i]=i*i;
    }
    if(K==1)work1();
    else printf("1

");
} 
/*
5 1
1 3 15 10 6

*/

k=2做法：

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

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

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<map>
#include<set>
#include<cstring>
#define MAXN 1000001
#define ps push_back
using namespace std;
int a[MAXN];
vector<int>v[650001];
int n,K;bool tong[MAXN];
bool biao[MAXN];
int ans[MAXN];
int poww[MAXN];
int last[MAXN];
void work1()
{
        int p=n;
        for(int i=n;i>=1;i--)
        {
            for(int j=512;j>=1;j--)
            {
                if(poww[j]<=a[i]) break;
                if(last[poww[j]-a[i]])
                {
                    for(int k=p;k>i;k--) 
                        last[a[k]]=0;
                    ans[++ans[0]]=i,p=i;
                    break;
                }
            }
            last[a[i]]++;
        }
        printf("%d
",ans[0]+1);
        for(int i=ans[0];i>=1;i--) 
            printf("%d ",ans[i]);
        cout<<endl;
}
int col[MAXN];
bool DFS(int x,int color,int r,int l,int fa)
{
    col[x]=color;
    for(int i=0;i<v[x].size();i++)
    {
        int to=v[x][i];
        if(to==fa||to>r||to<l)continue;
        if(col[to]==0)
        {
            if(!DFS(to,3-color,r,l,x))
            return 0;
        }
        if(col[to]==col[x])
        {
             return 0;
        }
    }
    return 1;
}
signed main()
{ 
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
    }  
    for(int i=1;i<=522;++i)
    {
        poww[i]=i*i;
        tong[i*i]=1;
    }
    if(K==1)work1();
    else 
    {
      int tail=n;
      for(int end=n;end>=1;--end)
      {
         for(int i=tail;i>end;i--)
         {
              if(tong[a[i]+a[end]])
              {
                 v[i].ps(end);v[end].ps(i);
              }
         }
         if(!DFS(end,1,tail,end,0))
         {
              ans[0]++;
              ans[ans[0]]=end;
              tail=end;
              v[end].clear();
         }
         for(int i=tail;i>=end;i--)
         {
             col[i]=0;
         }
      }
      printf("%d
",ans[0]+1);
      if(ans[0]==0)cout<<endl;
      for(int i=ans[0];i>=1;--i)
      {
          printf("%d ",ans[i]);
      }   
    }   
}