这种东西小一点的范围应该是合并果子,说的科学一点就是k叉哈夫曼树
容易易发现 k 叉哈夫曼树只有 n/k 个节点,并且哈夫曼树合并过程中合并出的节点权值是单调不不降的。
我们开两个队列,用一个队列列保存新合并出的节点。
每次取出前 k 小的权值,怎么选呢?我们枚举从第二个队列中选择i个,那么从第一个队列就是选了num-i个,可以选择哪个更优。
选择次数的复杂度是:n/1+n/2+n/3+…+n/k=n log n
那么枚举的复杂度呢?我们发现,如果最坏情况从第二个队列里选k个,只塞回去1个。所以就是个常数,加了快读就能过了
#include
#include
#include
#define LL long long
using namespace std;
const int N=1000005;
int n;LL a[N],b[N],s[N];
int read()
{
int ret = 0; char k = getchar();
while (k < '0' || k > '9') k = getchar();
while (k >= '0' && k <= '9') ret = ret * 10 + k - '0', k = getchar();
return ret;
}
LL Hafuman(int k)
{
int ai=1,bi=0,blen=0;
LL cost=0;
bool first=true;
while (ai<=n || blen-1>bi)
{
int num = 0;
if (first)
{
if ((n-k)%(k-1)==0) num = k;
else num=(n-k)%(k-1)+1;
first = false;
}
else num=k;
LL sum=0;int i,tmp=bi;
for (i=0;i<=num;i++)
{
int hd=num-i;
if (hd+ai-1<=n && (i+tmp>=blen || a[hd+ai-1]1])) break;
sum+=b[++bi];
}
sum+=s[num-i+ai-1]-s[ai-1]; ai+=num-i;
cost+=sum; b[++blen]=sum;
}
return cost;
}
int main()
{
freopen("epic.in","r",stdin);
freopen("epic.out","w",stdout);
scanf("%d",&n);
for (int i=1;i<=n;i++) a[i]=read();
sort(a+1,a+n+1);
for (int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
LL ans=0;
for (int i=2;i<=n;i++) ans^=Hafuman(i);
printf("%lld",ans);
}
我当然写了一个优美的42pts暴力
正解嘛。这是一个神网络流。
注意到满足条件的矩阵都符合所有相邻的点对中有且仅有 2 对颜色不同。
不考虑染色而是考虑分配任意两个相邻点对的值是否相同,会发现满足任意 2×2 的子矩阵中有且仅有 2 对颜色不同的情况下,必然对应着一组合法解。
在把问题这样转化以后考虑网络流,把所有 2×2 的子矩阵建成一个点,这样两个子矩阵相交的部分(我们只考虑相交两个点的而不是一个点)就是可能给他们带来贡献的点对,这样原图中两个相邻子矩阵相交部分的【不相等】可以转化成新图中的一条边,对新图黑白染色后就可以跑了,如果满流就是YES,否则NO。
注意我们要一开始剔除那些不可能合法的矩阵,即我们一开始在外面填上一圈1比较方便(填的原因是外面的一圈也是可以和矩阵的矩形有相交两点的)
42pts
#include
#include
#include
using namespace std;
const int N=35;
char st[N][N];int a[N][N],n;bool fff=0;
int dzmsls(int x,int y)
{
if (a[x-1][y-1]==0 && a[x-1][y]==0 && a[x][y-1]==0) return 1;
if (a[x-1][y-1]==0 && a[x-1][y]==1 && a[x][y-1]==1) return 1;
if (a[x-1][y-1]==1 && a[x-1][y]==0 && a[x][y-1]==0) return 0;
if (a[x-1][y-1]==1 && a[x-1][y]==1 && a[x][y-1]==1) return 0;
return -1;
}
void dfs(int id)
{
if (id>n*n){fff=1;return;}
int x=(id-1)/n+1,y=(id-1)%n+1;
if (a[x][y]==-1)
{
if (st[x][y]==st[x][y-1]) a[x][y]=a[x][y-1];
if (st[x][y]==st[x-1][y]) a[x][y]=a[x-1][y];
int must=dzmsls(x,y);
if (a[x][y]!=-1 && must!=-1 && a[x][y]!=must) return;
if (a[x][y]==-1) a[x][y]=must;
if (a[x][y]==-1)
for (int i=0;i<=1 && !fff;i++) a[x][y]=i,dfs(id+1),a[x][y]=-1;
else dfs(id+1);
}else
{
if (y>=2 && x>=1)
{
int must=dzmsls(x,y);
if (must!=-1 && a[x][y]!=must) return;
}
dfs(id+1);
}
}
int main()
{
freopen("color.in","r",stdin);
freopen("color.out","w",stdout);
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%s",st[i]+1);
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++) a[i][j]=-1;
for (int i=1;i<=n;i++) a[i][1]=st[i][1]-'0',a[i][n]=st[i][n]-'0',a[1][i]=st[1][i]-'0',a[n][i]=st[n][i]-'0';
dfs(n+2);
if (fff) printf("YES");else printf("NO");
}
100pts
#include
#include
#include
#include
using namespace std;
const int N=1000;
const int M=10000;
#define INF 1e9
int tot,nxt[M],point[N],remind[M],dis[N],v[M],id[100][100];bool vis[N];
char f[100][100],st[100][100];
void addline(int x,int y,int c)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; remind[tot]=c;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; remind[tot]=0;
}
bool bfs(int s,int t)
{
queue<int>q;
q.push(s);memset(dis,0x7f,sizeof(dis));
dis[s]=0;
memset(vis,0,sizeof(vis));
while (!q.empty())
{
int now=q.front(); q.pop(); vis[now]=0;
for (int i=point[now];i!=-1;i=nxt[i])
if (dis[v[i]]>dis[now]+1 && remind[i])
{
dis[v[i]]=dis[now]+1;
if (!vis[v[i]]) q.push(v[i]),vis[v[i]]=1;
}
}
return dis[t]int dfs(int now,int t,int limit)
{
if (now==t || !limit) return limit;
int flow=0,f;
for (int i=point[now];i!=-1;i=nxt[i])
if (dis[v[i]]==dis[now]+1 && remind[i] && (f=dfs(v[i],t,min(limit,remind[i]))))
{
flow+=f; limit-=f;
remind[i]-=f; remind[i^1]+=f;
if (!limit) break;
}
return flow;
}
int dinic(int s,int t)
{
int ans=0;
while (bfs(s,t))
ans+=dfs(s,t,INF);
return ans;
}
int main()
{
freopen("color.in","r",stdin);
freopen("color.out","w",stdout);
memset(point,-1,sizeof(point));tot=-1;
int n,m;scanf("%d",&n);
m=n+2;
for (int i=0;i<=m;i++) for (int j=0;j<=m;j++) f[i][j]='1';
for (int i=1;i<=n;i++) scanf("%s",st[i]+1);
for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) f[i][j]=st[i][j];
for (int i=0;i1;i++)
for (int j=0;j1;j++)
id[i][j]=i*(m-1)+j;
int s=(m-1)*(m-1),t=s+1,mx=0;
for (int i=0;i1;i++) for (int j=0;j1;j++)
{
int cnt=0;
if (f[i][j]=='1') cnt++;
if (f[i][j+1]=='1') cnt++;
if (f[i+1][j]=='1') cnt++;
if (f[i+1][j+1]=='1') cnt++;
if (cnt==4) continue;
mx++;
if ((i+j)&1) addline(s,id[i][j],2);else {addline(id[i][j],t,2);continue;}
if (f[i][j]!=f[i][j+1] && i-1>=0) addline(id[i][j],id[i-1][j],1);
if (f[i][j]!=f[i+1][j] && j-1>=0) addline(id[i][j],id[i][j-1],1);
if (f[i+1][j]!=f[i+1][j+1] && i+11][j],1);
if (f[i][j+1]!=f[i+1][j+1] && j+11],1);
}
if (dinic(s,t)==mx) printf("YES");else printf("NO");
}