[JZOJ5972]wang

Description

定义多项式F(x)是好的，当且仅当满足$F(2\ast F(x)-x+1)=F(x)+C$
现在有n个二元组(xi,yi)，需要最小化∑|F(xi)-yi|
n<=10000,C<=60

Solution

考虑怎样一个多项式是好的
性质一：若F(a)=t，a+b=2t+1，则F(b)=t+C
性质二：F(x+2C)=F(x)+2C
令y=2F(x)-x+1,则F(y)=F(x)+C
又F(2F(y)-y+1)=F(y)+C=F(x)+2C
2F(y)-y+1=2F(y)-2F(x)+x=x+2C
那么我们可以把所有的x按模2C分组，每组只需要确定一个组，整组就确定了
然后，组与组之间会存在相对的制约关系，并且根据性质一，只有奇偶性不同的组会产生关联，这显然是一个二分图，如果我们能求出所有的组(a,b)之间互相匹配的代价那么就是一个最小权匹配
现在问题变成了如何求出a和b匹配的代价
设a=2i,b=2j+1
那么需要满足F(a)=i+j+nC，C是正整数，然后我们就能把a和b组中所有的F(x)都求出来
简单推一下可以发现，若F(a)+=nC，则F(b)-=nC
问题变成了求一个n，使得代价最小
对于一个二元组(xi,yi)，如果在a组，那么设Wi=(i+j)+(xi-a)-y，否则Wi=-((i+j)+(xi-b)-y)
那么代价就可以统一成∑|Wi+nC|，求最小值显然排序求中位数
然后就没有了

Code

#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a) for(int i=lst[a];i;i=nxt[i])
using namespace std;

typedef long long ll;

const int N=1e5+5;
const ll INF=1e15;

int t[N],nxt[N],lst[N],f[N],l;
ll v[N];
void add(int x,int y,ll z) {
	t[++l]=y;f[l]=1;v[l]=z;nxt[l]=lst[x];lst[x]=l;
	t[++l]=x;f[l]=0;v[l]=-z;nxt[l]=lst[y];lst[y]=l;
}

vector<pair<int,int> > vec[N];
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)

int C,n,x[N],y[N],tot,a,b,P,S,T;
int q[N],pre[N];
bool vis[N];
ll dis[N],p[N],ans;

bool spfa() {
	fo(i,0,T) dis[i]=INF;dis[S]=0;
	fo(i,0,T) vis[i]=0;vis[S]=1;
	int i=0,j=1;q[1]=S;
	while (i<j) {
		rep(k,q[++i])
			if (f[k]&&dis[t[k]]>dis[q[i]]+v[k]) {
				dis[t[k]]=dis[q[i]]+v[k];
				pre[t[k]]=k;
				if (!vis[t[k]]) vis[q[++j]=t[k]]=1;
			}
		vis[q[i]]=0;
	}
	return dis[T]<INF;
}

void find() {
	int flow=1;
	for(int i=T;i!=S;i=t[pre[i]^1]) flow=min(flow,f[pre[i]]);
	for(int i=T;i!=S;i=t[pre[i]^1]) {
		f[pre[i]]-=flow;
		f[pre[i]^1]+=flow;
		ans+=flow*v[pre[i]];
	}
}

ll calc(ll x) {
	ll res=0;
	fo(i,1,tot) res+=abs(p[i]-x);
	return res;
}

int main() {
	freopen("wang.in","r",stdin);
	freopen("wang.out","w",stdout);
	scanf("%d%d",&C,&n);l=1;S=2*C;T=2*C+1;
	scanf("%d%d%d%d",&x[1],&a,&b,&P);
	fo(i,2,n) x[i]=((ll)x[i-1]*a%P+b)%P;
	scanf("%d%d%d%d",&y[1],&a,&b,&P);
	fo(i,2,n) y[i]=((ll)y[i-1]*a%P+b)%P;
	fo(i,1,n) vec[x[i]%(2*C)].pb(mp(x[i],y[i]));
	fo(i,0,C-1) add(S,2*i,0);
	fo(i,0,C-1) add(2*i+1,T,0);
	fo(i,0,C-1)
		fo(j,0,C-1) {
			int a=2*i,b=2*j+1;tot=0;
			for(int w=0;w<vec[a].size();w++) {
				int x=vec[a][w].first,y=vec[a][w].second;
				p[++tot]=(i+j)+(x-a)-y;
			}
			for(int w=0;w<vec[b].size();w++) {
				int x=vec[b][w].first,y=vec[b][w].second;
				p[++tot]=-((i+j)+C+(x-b)-y);
			}
			if (!tot) {add(a,b,0);continue;}
			sort(p+1,p+tot+1);
			if (tot&1) {
				ll mid=p[(tot+1)/2];
				ll now=mid-mid%C-((mid<0)?C:0);
				if (calc(now)>calc(now+C)) now+=C;
				add(a,b,calc(now));
			} else {
				ll mid=(p[tot/2]+p[tot/2+1])/2;
				ll now=mid-mid%C-((mid<0)?C:0);
				if (calc(now)>calc(now+C)) now+=C;
				add(a,b,calc(now));
			}
		}
	while (spfa()) find();
	printf("%lld\n",ans);
	return 0;
}