GalaxyOJ-951 (思维)

题目

Problem 951: 多愁善感
Time Limit: 2000 ms Memory Limit: 128000 KB

Problem Description

多愁善感的少年WSW要走过一条斑马线，斑马线是由n条交替的黑条和白条构成的，第一条是黑条。WSW脚的长度是s。WSW要求在走的过程中，他脚的任何一部分都不能碰到象征邪恶的黑条。第一条之前和第n条之后的部分都是白色的，WSW可以任意选择第一条之前的位置出发。但出发位置一旦选定，之后WSW的每一步的长度都必须是k。请你判断WSW有没有可能在不碰到黑条的情况下通过斑马线，即走到第n条之后。

Input

输入文件emotional.in有多组数据。
第一行一个整数t 表示数据组数。
每组数据的第一行有三个整数s, k, n。
第二行有n个整数A_1, A_2, …, A_n，依次表示黑白条的长度。

100%的数据，2<=n<=500000, 1<=s<k<=10^9, 1<=A_i<=10^9, 1<=t<=10。

Output

每组数据输出一行。
若能通过输出”TAK”，否则输出”NIE”。

Sample Input

6

3 9 10
3 3 3 1 1 3 2 1 2 1

3 13 5
3 2 1 3 3

3 12 8
1 1 3 3 2 2 1 1

3 9 6
3 3 1 1 3 1

2 8 7
2 5 6 3 2 1 2

2 8 4
1 6 7 4

Sample Output

NIE
TAK
TAK
TAK
TAK
NIE

分析

• 把线段看成数轴上紧贴着的线段，从0开始。
• 由于每步的长度是固定的，对于每个黑色区域，区间中所有数 %k ，映射到区间 [0,k-1]上，代表模数为这些的点都不能走到（不然肯定会踏上黑色区域）。最后看看[0,k-1]里有没有白色（没被染黑）的点，就知道有没有方案了。

程序

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
struct zzk{ll l,r;} a[1000005];     //区间 l,r 不可行 
bool cmp(zzk x,zzk y){
    if (x.l==y.l) return x.r>y.r;
    return x.l<y.l;
}
ll i,t,n,s,k,l,r,len,F,kl,kr,k1,k2,num,R;

void upd(ll x,ll y){
    if (y<x){return;}
    kl=x/k,kr=y/k;
    k1=x%k,k2=y%k;
    if (y-x>=k){a[++num]=(zzk){0,k-1}; return;}
    if (kl==kr){
        a[++num]=(zzk){k1,k2};
        return;
    }
    if (kr-kl==1){
        a[++num]=(zzk){k1,k-1};
        a[++num]=(zzk){0,k2};
        return;
    }
}

int main(){
    for (scanf("%lld",&t),F=1; t--; F=1,num=0){
        scanf("%lld%lld%lld",&s,&k,&n); //一步长 k，脚长 s 
        for (i=l=0; i<n && F; i++){
            scanf("%lld",&len);
            r=l+len,l+=1;   //右端点在 [l,r] 都不行 
            if (~i&1) upd(l,r+s-1);
            l=r;
        }
        sort(a+1,a+num+1,cmp);
        if (a[1].l>0){puts("TAK"); continue;}
        for (i=1,F=0,R=0; i<=num && !F; i++){
            if (a[i].l>R+1){F=1; continue;}
            R=max(R,a[i].r);
            while (i<num && a[i+1].l==a[i].l) i++;
        }
        if ((R<k-1) || F){puts("TAK"); continue;} 
        puts("NIE");
    }
}

提示

• 哇，测试时函数传参开成 int ，结果 WA 爆零……