srm 526 T1 && T2

T1:

problem:

给定一个n * m的矩阵，每行每列都至多有一个1，其余都是0。现在可以将某些1移动到某些0的位置上，代价：(a,b)->(c,d) : |a - c| + |b - d|。问将所有的1移动到某行上或者某列上且所有的1的位置是连续的最小代价。

solution:

枚举移动到哪行，把所有1先移动到这一行，这样就是一维的问题了。再枚举起点是哪个格子，最后贪心放即可。

T2:

problem:

有n个石子，两个人玩游戏。第一个人可以把石子取走1~K个，且剩下的石子个数是质数。第二个人可以把石子取走1~K个，且剩下的石子个数是偶数。两人的目标是一样的：若有必胜策略，则最小化游戏执行次数，否则最大化。问游戏最终执行了多少次。

solution:

设f[i][k] ，代表有i个石子，当前是第k个人操作，游戏执行的次数。暴力一点的做法是，枚举取走1~k，先检查是否有必胜策略，然后再判断是取最大还是取最小。但是由于K是固定的，所以就可以用单调队列来优化转移过程。

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 500000;

pair<int, int> stack0[2][N], stack1[2][N];
int p[2], q[2], t[2], r[2];
int f[N][2], len;
bool isprime[N];

inline void stapop(pair<int, int> stack[], int &p, int &t, int i) {
    while (p <= t && stack[p].second + len < i)
	++p;
}

inline void stapush(pair<int, int> stack[], int &p, int &t, pair<int, int> x) {
    while (p <= t && stack[t].first <= x.first)
	--t;
    stack[++t] = x;
}

class PrimeCompositeGame {
public:
    int theOutcome(int n, int K) {
	if (n < 2) return 0;
	len = K;
	memset(isprime, true, sizeof(isprime));
	for (int i = 2; i * i < N; ++i)
	    if (isprime[i]) 
		for (int j = i * i; j < N; j += i)
		    isprime[j] = false;
	p[0] = p[1] = q[0] = q[1] = 1; 
	for (int i = 2; i <= n; ++i) {
	    for (int j = 0; j < 2; ++j)
		stapop(stack0[j], p[j], t[j], i), stapop(stack1[j], q[j], r[j], i);
	    if (q[0] <= r[0]) f[i][0] = -stack1[0][q[0]].first + 1;
	    else {
		f[i][0] = 0;
		if (q[1] <= r[1]) f[i][0] = -stack1[1][q[1]].first - 1;
	    }
	    if (p[0] <= t[0]) f[i][1] = -stack0[0][p[0]].first + 1;
	    else {
		f[i][1] = 0;
		if (p[1] <= t[1]) f[i][1] = -stack0[1][p[1]].first - 1;
	    }
	    if (!isprime[i]) {
	    	if (f[i][0] <= 0) stapush(stack0[0], p[0], t[0], make_pair(f[i][0], i));
	    	else stapush(stack0[1], p[1], t[1], make_pair(f[i][0], i));
	    }
	    if (isprime[i]) {
	    	if (f[i][1] <= 0) stapush(stack1[0], q[0], r[0], make_pair(f[i][1], i));
	    	else stapush(stack1[1], q[1], r[1], make_pair(f[i][1], i));
	    }
	}
	return f[n][0];
    }
};